pandas dataframe 컬럼비교

예제1

import pandas as pd

# 예시 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9], 'D': [10, 11, 12], 'E': [13, 14, 15]})

# 확인할 컬럼명 리스트
my_columns1 = ['A', 'B', 'C', 'D']
my_columns2 = ['A', 'B', 'C']

# DataFrame의 컬럼명 리스트
df_cols = df.columns.tolist()

# if-elif 조건 사용
if all(col in df_cols for col in my_columns1):
    print('OK1')
elif all(col in df_cols for col in my_columns2):
    print('OK2')
else:
    print('Not OK')

OK1

예제2

import pandas as pd

# 예시 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9], 'E': [13, 14, 15]})

# 확인할 컬럼명 리스트
my_columns1 = ['A', 'B', 'C', 'D']
my_columns2 = ['A', 'B', 'C']

# DataFrame의 컬럼명 리스트
df_cols = df.columns.tolist()

# if-elif 조건 사용
if all(col in df_cols for col in my_columns1):
    print('OK1')
elif all(col in df_cols for col in my_columns2):
    print('OK2')
else:
    print('Not OK')

OK2

예제3

import pandas as pd

# 예시 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'E': [7, 8, 9], 'F': [13, 14, 15]})

# 확인할 컬럼명 리스트
my_columns1 = ['A', 'B', 'C', 'D']
my_columns2 = ['A', 'B', 'C']
my_columns3 = ['A', 'B']

# DataFrame의 컬럼명 리스트
df_cols = df.columns.tolist()

# if-elif 조건 사용
if all(col in df_cols for col in my_columns1):
    print('OK1')
elif all(col in df_cols for col in my_columns2):
    print('OK2')
elif all(col in df_cols for col in my_columns3):
    print('OK3')
else:
    print('Not OK')

OK3