파이썬 argmax 이해

매트릭스가 있다고 가정 해 봅시다.

import numpy as np    
A = np.matrix([[1,2,3,33],[4,5,6,66],[7,8,9,99]])

내가 아는 한 가장 큰 값을 반환하는 한 argmax 함수를 이해하려고합니다.

Python에서 시도한 경우 :

np.argmax(A[1:,2])

행의 끝 (세 번째 행)과 세 번째 열을 따라 두 번째 행에서 가장 큰 요소를 가져와야합니까? 그래서 그것은 배열 [6 9]이어야하고, arg max는 9를 반환해야합니까? 하지만 파이썬에서 실행할 때 왜 값 1을 반환합니까?

그리고 열 3 (9)의 행 2에서 가장 큰 요소를 반환하려면 코드를 어떻게 수정해야합니까?

Python 문서를 확인했지만 여전히 약간 불분명합니다. 도움과 설명에 감사드립니다.

 

해결 방법

 

argmax 없음은 가장 큰 값의 위치 를 반환합니다. max 는 가장 큰 값을 반환합니다.

import numpy as np    
A = np.matrix([[1,2,3,33],[4,5,6,66],[7,8,9,99]])

np.argmax(A)  # 11, which is the position of 99

np.argmax(A[:,:])  # 11, which is the position of 99

np.argmax(A[:1])  # 3, which is the position of 33

np.argmax(A[:,2])  # 2, which is the position of 9

np.argmax(A[1:,2])  # 1, which is the position of 9

 

참조 페이지 https://stackoverflow.com/questions/36300334