Numpy 배열에서 최대값 인덱스를 얻는 방법

다음 방법을 사용하여 NumPy 배열의 최대값 인덱스를 가져올 수 있습니다.

방법 1: 1차원 배열에서 최대값의 인덱스 가져오기

 x. argmax ()

방법 2: 다차원 배열의 각 행에서 최대값의 인덱스를 가져옵니다.

 x. argmax (axis= 1 )

방법 3: 다차원 배열의 각 열에서 최대값의 인덱스를 가져옵니다.

 x. argmax (axis= 0 )

다음 예에서는 각 방법을 실제로 사용하는 방법을 보여줍니다.

예시 1: 1차원 배열에서 최대값의 인덱스 가져오기

다음 코드는 1차원 NumPy 배열에서 최대값의 인덱스를 얻는 방법을 보여줍니다.

 import numpy as np

#create NumPy array of values
x = np. array ([2, 7, 9, 4, 4, 6, 3])

#find index that contains max value
x. argmax ()

2

argmax() 함수는 2 값을 반환합니다.

이는 배열의 인덱스 위치 2 에 있는 값에 최대값이 포함되어 있음을 알려줍니다.

원본 배열을 보면 인덱스 위치 2 의 값이 9 임을 알 수 있는데, 이는 실제로 배열의 최대값입니다.

예시 2: 다차원 배열의 각 행에서 최대값의 인덱스 가져오기

다음 코드는 다차원 NumPy 배열의 각 행에서 최대값의 인덱스를 얻는 방법을 보여줍니다.

 import numpy as np

#create multi-dimentional NumPy array
x = np. array ([[4, 2, 1, 5], [7, 9, 2, 0]])

#view NumPy array
print (x)

[[4 2 1 5]
 [7 9 2 0]]

#find index that contains max value in each row
x. argmax (axis= 1 )

array([3, 1], dtype=int32)

결과에서 다음을 확인할 수 있습니다.

• 첫 번째 줄의 최대값은 인덱스 위치 3 에 있습니다.
• 두 번째 줄의 최대값은 인덱스 위치 1 에 있습니다.

예시 3: 다차원 배열의 각 열에서 최대값의 인덱스 가져오기

다음 코드는 다차원 NumPy 배열의 각 열에서 최대값의 인덱스를 얻는 방법을 보여줍니다.

 import numpy as np

#create multi-dimentional NumPy array
x = np. array ([[4, 2, 1, 5], [7, 9, 2, 0]])

#view NumPy array
print (x)

[[4 2 1 5]
 [7 9 2 0]]

#find index that contains max value in each column
x. argmax (axis= 0 )

array([1, 1, 1, 0], dtype=int32)

결과에서 다음을 확인할 수 있습니다.

• 첫 번째 열의 최대값은 인덱스 위치 1 에 있습니다.
• 두 번째 열의 최대값은 인덱스 위치 1 에 있습니다.
• 세 번째 열의 최대값은 인덱스 위치 1 에 있습니다.
• 네 번째 열의 최대값은 인덱스 위치 0 에 있습니다.

관련 항목:NumPy 축에 대한 간단한 설명

추가 리소스

다음 튜토리얼에서는 Python에서 다른 일반적인 작업을 수행하는 방법을 설명합니다.

NumPy 배열을 값으로 채우는 방법
NumPy 배열의 요소를 바꾸는 방법
NumPy 배열에서 특정 행을 얻는 방법