조건문으로 파생변수 만들기

1. conditon 변수를 생성

 - condition에 따라 True, False가 출력됨

condition = X['wt'] < 1

 

2. wt_class라는 컬럼을 만들고 조건을 만족할 경우 True를 0으로 불만족할 경우 1로 저장하는 파생변수 생성

X.loc[condition, 'wt_class'] = 0

X.loc[~condition, 'wt_class'] = 1

 

3. 출력값에 이상이 없으면 wt컬럼은 삭제

X = X.drop(columns = ['wt'])

 

 

# 연산을 통한 파생컬럼은 

X['qsec_3'] = X['qsec'] * 3