Как выбрать из 2D numpy .array где столбец == условие

Question

В качестве примера, у меня есть набор записей в 2D numpy .array, и я хотел бы выбрать все записи, где значение в 3-м столбце равно 10. Есть ли способ сделать это, кроме циклического прохождения массив и построить список выбранных записей?

Answer 1

Как только я узнал о концепции «фильтра», я искал еще немного и нашел ответ, который искал в этом stackoverflow-вопросе .

Итак, в примере с автомобилем фильтр будет записано как df [df [:, 1] == 'машина']

Answer 2

Здесь я создал массив numpy.

print(df)

o / p is: array ([['A', 'car'], ['B') , 'bike'], ['C', 'car'], ['D', 'truck']], dtype = object)

Теперь, чтобы выбрать все строки с автомобилем, вы можете просто Отфильтруйте это значение из массива

df[df == "car"]

o / p: массив (['car', 'car'], dtype = object)

Если вы хотите, чтобы он был преобразован в список python, оберните оператор в списке, то есть

list(df[df == "car])