[PYTHON] pandas query 함수 사용하기

0. 목표

 - query 함수 사용하기

1. 실습하기

 1) library 호출

import pandas as pd

 2) dataframe 생성

df = pd.DataFrame({'A' : [1, 2, 3, 4, 5], 'B' : ['apple', 'banana', 'apple', 'berry', 'watermelon']})

 3) query 사용하기

  (1) where A > 3

df.query('A > 3')

  (2) where A > 3 and A < 5

df.query('A > 3 and A < 5')

  (3) where A = 1 or A = 4

df.query('A == 1 or A == 4')

  (4) where A in (1, 3, 5)

df.query("A in (1, 3, 5)")

  (5) where A not in (1, 3, 5)

df.query('A not in (1, 3, 5)')

  (5) where B like '%b%'

df.query("B.str.contains('b')") # 에러 발생시 df.query("B.str.contains('b')", engine='python')

2. 참고

 - [Pandas] Query 함수 사용법 (Pandas의 꽃)

 - pandas.DataFrame.query

 - pandas Python Series objects are mutable, thus they cannot be hashed in query method