Пропустить строки при чтении данных из файла с помощью 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) 

Answer 2

Я тоже хотел эту функцию, но ее не существует. Поэтому на данный момент вы, вероятно, должны просто загрузить весь файл в виде матрицы, затем использовать причудливую индексацию для подмножества или удалить части, которые не хотите, и т. Д. Если количество строк, которое вы действительно хотите, очень мало, а общий размер очень велика, вам даже может быть лучше просто перебирать строки в обычном Python.

Answer 3

Вы можете прочитать весь файл и фильтр затем с помощью фантазии индексации:

a = np.genfromtxt('test.txt', dtype=str) 

check = (a[:,0]=='A') 
a = a[check]