pandas dataframe KeyError 오류 해결하기

판다스 DataFrame을 사용할 때 KeyError가 발생하는 예시 코드와 그 해결 방법을 설명하겠습니다.

예시 코드:

import pandas as pd

# 샘플 데이터프레임 생성
data = {'이름': ['Alice', 'Bob', 'Charlie'],
        '나이': [30, 25, 35]}

df = pd.DataFrame(data)

# 존재하지 않는 열에 접근
print(df['키'])  # KeyError 발생

위 코드에서는 '키'라는 열 이름이 데이터프레임에 존재하지 않기 때문에 KeyError가 발생합니다. 이러한 오류를 해결하려면 다음과 같은 방법을 사용할 수 있습니다.

해결 방법:

열 이름 확인:

먼저 데이터프레임에 존재하는 열 이름을 확인하세요. df.columns 속성을 사용하여 열 이름 목록을 확인할 수 있습니다.

print(df.columns)
# 출력: Index(['이름', '나이'], dtype='object')

올바른 열 이름 사용:

올바른 열 이름을 사용하여 접근하세요. 오탈자나 대소문자 구분에 주의하세요.

# '이름' 열에 접근
print(df['이름'])
# 출력: 0      Alice
#       1        Bob
#       2    Charlie
#       Name: 이름, dtype: object

열 이름으로 접근할 때 인덱스 사용:

열 이름이나 열의 위치(인덱스)를 사용하여 열에 접근할 수 있습니다.

# 인덱스 0의 열에 접근 (첫 번째 열)
print(df.iloc[:, 0])
# 출력: 0      Alice
#       1        Bob
#       2    Charlie
#       Name: 이름, dtype: object

열 존재 여부 확인:

열이 존재하는지 확인한 후 접근하세요.

if '키' in df.columns:
    print(df['키'])
else:
    print("열 '키'가 데이터프레임에 존재하지 않습니다.")

KeyError는 주로 열 이름을 잘못 사용하거나 존재하지 않는 열에 접근할 때 발생합니다. 열 이름을 정확하게 확인하고, 필요한 경우 인덱스를 사용하여 열에 접근하면 이 오류를 해결할 수 있습니다.