아미(아름다운미소)

assign 본문

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