python - 使用 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)