Как из двухмерного numpy массива получить индекс строки в которой есть цифра 2 без цикла?

Question

[Даниил Шевкунов]

есть массив например

a = np.array([[0, 1, 2], 
              [3, 4, 5],
              [6, 7, 8]])

Как получить индекс строки в которой есть цифра 2 без цикла?

Answer 1

[aRegius]

np.argwhere(a==2)[:, 0]