Найти номер строки, используя первый элемент строки

Question

У меня есть матрица numpy, где я храню какой-то ключ в первом элементе каждой строки (или иначе все ключи находятся в первом столбце).

[[123,0,1,1,2], 
[12,1,2,3,4], 
[1,0,2,5,4], 
[90,1,1,4,3]] 

Я хочу получить номер строки, поиск по ключевому слову. Я обнаружил, что для этого можно использовать numpy.where, но неясно, как использовать его для получения номера строки. Я хочу что-то вроде

>>numpy.func(myMatrix,90) 
3 

Любая помощь будет оценена.

Answer 1

Сравните первую колонку с 90 в пределах np.where. Она возвращает массив индексов элементов, равных с 90:

In [3]: A = np.array([[123,0,1,1,2], 
[12,1,2,3,4], 
[1,0,2,5,4], 
[90,1,1,4,3]]) 


In [6]: np.where(A[:,0]==90)[0] 
Out[6]: array([3]) 

Answer 2

Согласно online doc, numpy.where если вы даете это только логический массив будет возвращать список координат (один лист на измерение) для элементы, которые являются True. Итак, мы можем получить нужную информацию, захватив первый столбец вашего массива, сравнив его с элементом, который вы хотите найти, и вызовите np.where в этом булевом массиве. Все, что будет выглядеть так:

row,column=np.where(myMatrix[;,0]==90) 
#Note that column will just be 0 here