Python Pandas – 인덱스에서 최댓값의 정수 위치를 반환하는 방법(argmax)

Pandas 인덱스(Index)에서 가장 큰 값이 위치한 정수(int) 위치를 반환하려면 index.argmax() 메서드를 사용하면 됩니다. 이 메서드는 최댓값 자체가 아니라 해당 값이 있는 위치(인덱스 번호)를 반환한다는 점에 유의하세요.

기본 사용법

1. 필요한 라이브러리 임포트

import pandas as pd

2. 인덱스 생성

index = pd.Index([15, 25, 55, 10, 100, 70, 35, 40, 55])

3. 인덱스 출력하기

print("Pandas Index...\n", index)

4. 최댓값의 정수 위치 구하기

print("\n인덱스에서 최댓값의 정수 위치...\n", index.argmax())

전체 예제 코드

다음은 위 과정을 모두 포함한 완성된 코드입니다.

import pandas as pd

# 인덱스 생성
index = pd.Index([15, 25, 55, 10, 100, 70, 35, 40, 55])

# 인덱스 출력
print("Pandas Index...\n", index)

# 내부 데이터의 shape(형태) 튜플 반환
print("\n내부 데이터의 shape 튜플...\n", index.shape)

# 인덱스에 포함된 요소의 개수 반환
print("\n인덱스의 요소 개수...\n", index.size)

# 인덱스에서 최댓값의 정수 위치 반환
print("\n인덱스에서 최댓값의 정수 위치...\n", index.argmax())

실행 결과

위 코드를 실행하면 다음과 같은 출력 결과를 얻을 수 있습니다.

Pandas Index...
Int64Index([15, 25, 55, 10, 100, 70, 35, 40, 55], dtype='int64')

A tuple of the shape of underlying data...
(9,)

Number of elements in the index...
9

Get the int position of the largest value in the Index...
4

결과 해석

실행 결과를 보면 argmax() 메서드가 4를 반환했습니다. 이는 인덱스 요소 중 가장 큰 값인 100이 0부터 시작하는 위치 기준으로 네 번째(즉, 다섯 번째 요소)에 있음을 의미합니다. 즉, [15, 25, 55, 10, 100, 70, 35, 40, 55]에서 값 100의 위치가 4라는 뜻입니다.

참고 사항

• argmax()는 최댓값의 위치(position)를 반환하고, max()는 최댓값 자체를 반환합니다.
• 최댓값이 여러 개 존재하는 경우, 첫 번째로 나타나는 위치가 반환됩니다.
• NumPy 배열에도 동일한 이름의 argmax() 함수가 있으며, 동작 방식이 유사합니다.