행의 첫 번째 요소를 사용하여 행 번호 찾기

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에 비교하십시오. 만약 단지 부울 어레이는 좌표의리스트 (치수 하나씩 list)을 반환주는 경우 online doc 따르면 numpy.where을

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

: 그것은 90와 동일한 아이템의 인덱스의 배열을 반환 요소는 True입니다. 배열의 첫 번째 열을 찾아 원하는 요소와 비교하여 부울 배열에서 np.where를 호출하면 원하는 정보를 얻을 수 있습니다. 그 모든 모습은 다음과 같습니다 :

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