numpy.argmax¶

numpy.argmax 함수는 최대값을 갖는 성분의 인덱스를 반환합니다.

예제1¶

import numpy as np

a = np.array([0, 1, 4, 2, 3])
b = np.array([0, 1, 4, 2, 3, 4])

print(np.argmax(a))
print(np.argmax(b))

2
2

최대값이 여러 개 있다면 처음으로 발견한 성분의 인덱스만 반환합니다.

import numpy as np

a = np.array([[1, 2, 3], [4, 3, 2]])

print(np.argmax(a))

2차원 어레이의 경우 1차원으로 바꾼 후 인덱스를 반환합니다.