[pandas] 조건에 맞는 데이터 필터링 : where( ), query( )

🥑 조건에 맞지 않는 값을 일괄 대체하고 싶을 때 : where( )

• 코드 예제를 통해 확인 : num_actor 열의 값이 2이면 그대로 두고, 아닐 경우 False로 변경
• 형식 : .where(조건, 거짓일때의 값)

 

copied_data  = df.copy()
copied_data['two_actors'] = copied_data['num_actor'].where(copied_data['num_actor'] == 2, "False")
copied_data[['two_actors']].head(2)

 

 

🥑 코드를 효율적으로 작성하여 필터링하고 싶을 때 : query( )

• query( ) 메서드는 문자열로 표현된 조건을 입력으로 받는다
• 조건식은 데이터프레임의 열 이름을 사용하여 작성된다
• 여러개의 조건은 논리 연산자를 활용하여 작성할 수 있다
• 아래 예시에서, data[data['time']] < 60 보다 훨씬 간단하게 필터링이 가능

copied_data = df.copy()
under_60 = copied_data.query('time < 60')
under_60.head()

 

 

copied_data = df.copy()
under_2staff = copied_data.query('num_staff < 2')
under_2staff.head()