python numpy.where

이번에 시험을 봤다.

 

numpy.where을 알았더라면 편하게 했을건데.. 요건 잘 몰랐어서 그냥 condition 만들어서 target 데이터프레임만 뽑고

 

인덱스 뽑아내서 수정을 했다.

 

하지만 numpy.where을 사용했더라면 더 편리하게 가능했다.

 

numpy.where은 [condition, then x, else y] 식으로 사용이 가능하다.

 

예를 들어 국어 점수가 90점 이상이면 pass 아니면 fail 이라고 test 칼럼을 수정한다고 치자

 

그러면 다음과 같이 표현이 가능하다.

 

temp['test'] = numpy.where(temp['국어'] >= 90, 'pass', 'fail')