Np.where в матрице NumPy MxN, но возвращает M строк с индексами, в которых существует условие ⇐ Python

[Anonymous]

Я пытаюсь использовать np.where в цифровой матрице MxN, где я хочу вернуть то же количество строк M, но индексы в каждой строке, где существует элемент. Возможно ли это сделать? Например:
a = [[1 ,2, 2]
[2, 3, 5]]

np.where(a == 2)

Я бы хотел, чтобы это вернулось:
[[1, 2],
[0]]


Подробнее здесь: https://stackoverflow.com/questions/787 ... ndition-ex