Numpy получить индекс строки, где элементы в определенных столбцах равны нулю

Question

Я хочу найти индексы строки на основе критериев для определенных столбцов

Итак, что-то вроде:

import numpy as np
x = np.random.rand(4, 5)

x[2, 2] = 0
x[2, 3] = 0
x[3, 1] = 0
x[1, 3] = 0

Теперь я хочу получить индекс строк, где столбцы 3 или 4 являются нулями. Как можно сделать это с NumPy? Нужно ли делать несколько вызовов на nonzero для каждого столбца и объединять эти индексы, используя set или что-то в этом роде?

Answer 1

Использование np.where first array в строке tuple is index

np.where(x[:,[3,4]]==0)

Out[79]: (array([1, 2], dtype=int64), array([0, 0], dtype=int64))