numpy 배열에서 최대 값을 포함하는 행 또는 열을 찾습니다.

Question

배열 차원의 최대 값을 포함하는 행 또는 열을 두 번째 numpy 배열에서 찾는 방법은 무엇입니까?

Answer 1

에로 np.unravel_index와 함께 np.argmax를 사용할 수 있습니다 열의

np.argmax(np.max(x, axis=1)) 

을, 그리고에 대한

np.argmax(np.max(x, axis=0)) 

열.

Answer 2

np.where(x == np.max(x))을 사용할 수 있습니다. 예를 들어

:

>>> x = np.array([[1,2,3],[2,3,4],[1,3,1]]) 
>>> x 
array([[1, 2, 3], 
     [2, 3, 4], 
     [1, 3, 1]]) 
>>> np.where(x == np.max(x)) 
(array([1]), array([2])) 

제 값은 행 번호이고, 두 번째 숫자는 컬럼 번호이다.

Answer 3

당신은 하나 또는 다른이 필요하면 당신은

x = np.random.random((5,5)) 
print np.unravel_index(np.argmax(x), x.shape) 

Answer 4

np.argmax은 병합 된 배열에서 (가장 큰) 첫 번째 요소의 색인을 반환합니다. 따라서 배열의 모양을 알면 행/열 색인을 쉽게 찾을 수 있습니다.

A = np.array([5, 6, 1], [2, 0, 8], [4, 9, 3]) 
am = A.argmax() 
c_idx = am % A.shape[1] 
r_idx = am // A.shape[1]