我有一个数据集文件,如下所示:
A 1
B 2
C 3
A 4
B 5
C 6
我想提取只包含'A'
的行。如何使用numpy.genfromtxt
做到这一点?是否有其他在Numpy中可以帮助我完成同样任务的函数可用?
numpy.genfromtxt
可以接受生成器,因此您可以将genfromtext
和ifilter
链接在一起:
from itertools import ifilter
with open(fname, 'rb') as inp:
filtered_inp = ifilter(lambda x: x.startswith('A'), inp)
a = np.genfromtxt(filtered_inp)
a = np.genfromtxt('test.txt', dtype=str)
check = (a[:,0]=='A')
a = a[check]