Python 판다스에서 열이 존재하는지 확인하는 방법은 무엇인가요?, How to check if a column exists in Pandas

질문

판다스 데이터프레임 df에서 열이 존재하는지 확인하는 방법은 무엇인가요?

   A   B    C
0  3  40  100
1  6  30  200

위의 데이터프레임에서 열 "A"가 존재하는지 확인하려면 어떻게 해야 합니까? 그래서 다음을 계산할 수 있습니다:

df['sum'] = df['A'] + df['C']

그리고 "A"가 존재하지 않는다면:

df['sum'] = df['B'] + df['C']

답변

이것은 작동합니다:

if 'A' in df:

하지만 명확하게 작성하기 위해 아마 다음과 같이 작성할 것입니다:

if 'A' in df.columns: