[python] conditional to generate new column조건 걸고 새 칼럼 만들기

fig1. title

파이썬에서 조건 걸고 새칼럼 추가하기

R에서는 mutate + ifelse 로 간단하게 가능함

 

파이썬에서는 조건의 개수에 따라 편하게 쓰는 방법이 다른것 같음

 

---

#1. 조건이 1개일 때 : np.where

df2['eGFR_ab90'] = np.where(df2['eGFR_ckd']<90, 1, 0)

fig2. output of #1

#2. 조건이 2개 이상일때 : np.select

conditions = [
    (df_test['eGFR_ab'] == 1) & (df_test['pred'] == 1),
    (df_test['eGFR_ab'] == 1) & (df_test['pred'] == 0),
    (df_test['eGFR_ab'] == 0) & (df_test['pred'] == 1),
    (df_test['eGFR_ab'] == 0) & (df_test['pred'] == 0)
]

vals = ['TP', 'FN', 'FP', 'TN']

df_test['type'] = np.select(conditions, vals)

fig3. output of #2

 

 

참고 : www.dataquest.io/blog/tutorial-add-column-pandas-dataframe-based-on-if-else-condition/

Add a Column in a Pandas DataFrame Based on an If-Else Condition