np.where 을 사용한 조건문 column 추가

DataFrame에 if 문을 사용하여 새로운 column을 추가해야 한다.

=> np.where() 을 사용하여 쉽게 가능하다.

 

예를 들어, 다음과 같은 데이터 프레임이 있다고 해보자. real_test 와 prediction의 예측값이 같으면,

새로운 column을 추가하여 True라고 입력을 넣고, 아니면 False라고 입력을 넣는다고 하자.

다음과 같이 실행하면 된다.

=> np.where(조건, 조건에 해당하는 경우 넣을 값, 조건에 해당하지 않는 경우 넣을 값) 

confusion_df['일치'] = np.where(confusion_df['real_test'] == confusion_df['prediction'], True, False)

 

결과는 아래와 같다.