pandas KeyError: 'non_existing_key' 오류 해결하기

소개

파이썬의 pandas 라이브러리를 사용할 때 종종 'KeyError: 'non_existing_key'' 에러를 보게 됩니다. 이 오류는 접근하려는 데이터프레임의 열이 존재하지 않을 때 발생하는데요, 주로 잘못된 열 이름이나 오타로 인해 발생하곤 합니다. 이 블로그 글에서는 이러한 에러의 원인과 해결 방법을 함께 살펴보겠습니다.

에러 발생 예시 코드

아래는 'KeyError: 'non_existing_key'' 에러가 발생할 수 있는 간단한 코드 예시입니다.

import pandas as pd
 
# 데이터프레임 생성
data = {'name': ['Alice', 'Bob'], 'age': [25, 30]}
df = pd.DataFrame(data)
 
# 존재하지 않는 열에 접근
print(df['non_existing_key'])

에러 해결 방법

1. 열 이름을 확인하기

먼저 데이터프레임에 어떤 열이 존재하는지 확인해야 합니다. 이를 통해 잘못된 열 이름을 수정할 수 있습니다.

import pandas as pd
 
# 데이터프레임 생성
data = {'name': ['Alice', 'Bob'], 'age': [25, 30]}
df = pd.DataFrame(data)
 
# 데이터프레임의 열 확인
print(df.columns)

2. 열 이름의 오타 수정

가끔씩 오타 때문에 발생하는 문제가 있으니, 확인한 열 이름을 바탕으로 접근해봅시다.

import pandas as pd
 
# 데이터프레임 생성
data = {'name': ['Alice', 'Bob'], 'age': [25, 30]}
df = pd.DataFrame(data)
 
# 올바른 열 이름으로 접근
print(df['name'])

3. 열 존재 여부를 확인한 후 접근하기

여러 개의 열에 접근해야 할 경우, try-except 문을 사용하여 예외 처리를 통해 안전하게 접근하는 것도 좋은 방법입니다.

import pandas as pd
 
# 데이터프레임 생성
data = {'name': ['Alice', 'Bob'], 'age': [25, 30]}
df = pd.DataFrame(data)
 
# 열 존재 여부 확인 후 접근
key = 'non_existing_key'
try:
    print(df[key])
except KeyError:
    print(f"'{key}' 열은 데이터프레임에 존재하지 않습니다.")

마무리

이 블로그 글에서는 pandas에서 발생하는 'KeyError: 'non_existing_key'' 에러를 해결하는 여러 가지 방법을 살펴보았습니다. 데이터프레임의 열 목록을 확인하고, 오타를 수정하며, 예외처리를 통해 보다 안전하게 데이터에 접근하는 방법을 알아보았습니다. pandas를 사용할 때는 항상 데이터프레임의 구조를 잘 파악하고 접근하는 것이 중요합니다!