ComputerLanguage_Program/PYTHON
[PYTHON] pandas query 함수 사용하기
pbj0812
2021. 7. 9. 10:14
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 Python Series objects are mutable, thus they cannot be hashed in query method