Finden Sie die Zeilennummer mit dem ersten Element der Zeile

Question

Ich habe eine numpy Matrix, in der ich eine Art von Schlüssel in das erste Element jeder Zeile speichern (oder auf andere Weise alle Schlüssel in der ersten Spalte sind).

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

Ich möchte die Zeilennummer durch den Schlüssel suchen bekommen. Ich fand, dass wir numpy.where dafür verwenden können, aber nicht klar, wie man es verwendet, um die Zeilennummer zu erhalten. Ich möchte etwas wie

>>numpy.func(myMatrix,90) 
3 

Jede Hilfe wäre willkommen.

Answer 1

Vergleichen Sie die erste Spalte mit 90 innerhalb np.where. Es wird ein Array mit den Indizes der Elemente zurück, die mit 90 gleich sind:

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

Nach dem online doc, numpy.where wenn Sie geben es nur ein boolean-Array-Listen von Koordinaten zurück (eine Liste pro Dimension) für die Elemente, die True sind. So können wir die gewünschten Informationen erhalten, indem Sie die erste Spalte Ihres Arrays abfangen, sie mit dem gewünschten Element vergleichen und np.where in diesem booleschen Array aufrufen. All das würde so aussehen:

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