랭귀지/pandas
pandas 컬럼리스트 확인
유키공
2024. 7. 25. 08:14
import pandas as pd
# 예시 DataFrame 생성
data1 = {
'a': [1, 2, 3],
'b': [4, 5, 6],
'c': [7, 8, 9]
}
data2 = {
'b': [1, 2, 3],
'c': [4, 5, 6],
'd': [7, 8, 9]
}
df1 = pd.DataFrame(data1)
df2 = pd.DataFrame(data2)
# 두 DataFrame의 컬럼 이름을 합치고 중복 제거
combined_columns = set(df1.columns).union(set(df2.columns))
# 결과를 리스트로 변환
unique_column_names = list(combined_columns)
# 결과 출력
print(unique_column_names)
함수
import pandas as pd
def get_unique_column_names(*dataframes):
"""
여러 개의 DataFrame에서 중복을 제거한 컬럼 이름 리스트를 반환하는 함수.
:param dataframes: DataFrame 객체들
:return: 중복이 제거된 컬럼 이름 리스트
"""
combined_columns = set()
for df in dataframes:
combined_columns.update(df.columns)
return list(combined_columns)
# 예시 DataFrame 생성
data1 = {
'a': [1, 2, 3],
'b': [4, 5, 6],
'c': [7, 8, 9]
}
data2 = {
'b': [1, 2, 3],
'c': [4, 5, 6],
'd': [7, 8, 9]
}
df1 = pd.DataFrame(data1)
df2 = pd.DataFrame(data2)
# 함수 호출
unique_columns = get_unique_column_names(df1, df2)
# 결과 출력
print(unique_columns)