랭귀지/SWIFT

pandas list 형태로 컬럼명 지정하고 list 컬럼에 접미사 _calc 일괄 붙여서 컬럼 rename 후 일괄 fillna 적용

유키공 2024. 6. 10. 22:09
import pandas as pd

# 예제 데이터프레임 생성
data = {
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
}
df = pd.DataFrame(data)

# 접미사를 붙일 컬럼 리스트 지정
columns_to_rename = ['A', 'B']

# 컬럼 이름 변경
df.rename(columns={col: col + '_calc' for col in columns_to_rename}, inplace=True)

# 해당 컬럼들에 대해 fillna(0) 적용
for col in columns_to_rename:
    df[col + '_calc'] = df[col + '_calc'].fillna(0)

# 결과 출력
print(df)