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