아미(아름다운미소)

a컬럼이 true이면 b값을 c에 넣어주고 아니몈 b값을 d에넣어주기 본문

랭귀지/pandas

a컬럼이 true이면 b값을 c에 넣어주고 아니몈 b값을 d에넣어주기

유키공 2024. 9. 11. 13:16
import pandas as pd

# 예시 DataFrame
data = {
    'a': [True, False, True, False],
    'b': [1, 2, 3, 4]
}
df = pd.DataFrame(data)

# a가 True이면 b를 c에, 아니면 b를 d에 넣기
df['c'] = df['b'].where(df['a'] == True)
df['d'] = df['b'].where(df['a'] == False)

# 결과 출력
print(df)
Comments