pandas TypeError: 'int' object is not subscriptable 오류 해결하기

소개

파이썬의 pandas 라이브러리를 사용하다보면 'TypeError: 'int' object is not subscriptable'라는 오류에 봉착할 수 있습니다. 이 오류는 주로 정수형(int) 객체를 리스트나 데이터프레임과 같은 서브스크립트 방식으로 접근하려 할 때 발생합니다. 이번 블로그 글에서는 이 오류의 원인과 해결 방법을 안내하겠습니다.

에러 발생 예시 코드

먼저, 이 오류가 발생할 수 있는 간단한 예시 코드를 살펴보겠습니다.

import pandas as pd

# 데이터프레임 생성
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)

# 정수형 변수를 설정
row_index = 1

# 잘못된 서브스크립트 접근
value = df[row_index][1]  # TypeError 발생
print(value)

에러 해결 방법

1. 올바른 데이터 접근 방법 사용

이러한 오류를 피하기 위해서는 정수형 인덱스를 사용하여 데이터프레임에 접근할 때 올바른 방법을 사용하는 것이 중요합니다. 아래 코드를 통해 각 열의 특정 값에 접근할 수 있습니다.

import pandas as pd

# 데이터프레임 생성
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)

# 올바른 서브스크립트 접근
value = df.iloc[row_index, 1]  # 이렇게 해야 합니다
print(value)

2. 데이터프레임의 모양 확인

데이터프레임에 대해 잘못된 인덱스를 사용하고 있을 수도 있습니다. 따라서 데이터프레임의 모양을 확인하는 것도 좋은 방법입니다.

import pandas as pd

# 데이터프레임 생성
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)

# 데이터프레임 모양 확인
print(df.shape)  # (3, 2) : 3행, 2열

# 안전하게 값에 접근하기
if 1 < df.shape[0] and 1 < df.shape[1]:
    value = df.iloc[1, 1]
    print(value)
else:
    print("인덱스가 데이터프레임의 범위를 벗어났습니다.")

마무리

이번 블로그 글에서는 pandas에서 발생할 수 있는 'TypeError: 'int' object is not subscriptable' 오류에 대해 알아보았습니다. 올바른 데이터 접근 방법인 iloc를 사용하거나 데이터프레임의 모양을 확인함으로써 이러한 오류를 피할 수 있습니다. pandas를 사용할 때는 항상 데이터 구조를 이해하고 적절한 메서드를 활용하여 코드의 효율성을 높이는 것이 중요합니다.