Если у меня есть Numpy массив так:Numpy: поиск первой совпадающей строки
import numpy as np
x = np.array([[0,1],[0,2],[1,1],[0,2]])
Как я могу вернуть индекс первой строки, которая соответствует [0,2]
?
Для списков это легко с помощью index
:
[[0,1],[0,2],[1,1],[0,2]]
l.index([0,2])
> 1
И я знаю, что numpy
имеет функцию numpy.where
, но я не совсем уверен, что сделать из выхода numpy.where
:
np.where(x==[0,2])
> (array([0, 1, 1, 3, 3]), array([0, 0, 1, 0, 1]))
Там также numpy.argmax
, но это также не возвращает то, что я ищу, что является просто индексом 1
np.argmax(x == [0,2], axis = 1)