랭귀지/pandas
assign
유키공
2025. 1. 9. 10:20
import pandas as pd
# 초기 데이터프레임 생성
df = pd.DataFrame({'ccc': [1, 2, 3]})
# 조건에 따라 추가할 열 정의
new_columns = {}
if df['ccc'].max() > 2:
new_columns['aaa'] = df['ccc']
if df['ccc'].min() < 2:
new_columns['bbb'] = df['ccc'] * 2
# assign 메서드에 **를 사용하여 조건부로 열 추가
df = df.assign(**new_columns)
print(df)
ccc aaa bbb
0 1 1.0 2
1 2 2.0 4
2 3 3.0 6