아미(아름다운미소)

pandas 컬럼리스트 확인 본문

랭귀지/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)
Comments