使用numpy.genfromtxt从文件读取数据时跳过行

Question

我在文件中有一个数据集,如下所示

A 1
B 2
C 3
A 4
B 5
C 6

我想只提取包含的行'A'.我该怎么做numpy.genfromtxt？numpy中是否有其他功能可以帮助我做同样的事情？

Answer 1

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)