본문 바로가기

Python/Data analysis

[Python] Data Analysis - DataFrame - Query

이번 시간에는 DataFrame 의 Query 에 대해서 알아보겠습니다.

 

 

df
----------------------------
	nm	open	close
cd			
A060310	3S	2920	2800
A095570	AJ네트웍스	1920	1900
A006840	AK홀딩스	2020	2010
A054620	APS홀딩스	3120	3200


cond = df['open'] >= 2000
df[cond]
----------------------------
	nm	open	close
cd			
A060310	3S	2920	2800
A006840	AK홀딩스	2020	2010
A054620	APS홀딩스	3120	3200

 

 

상기 내용을 Query 로 표현 하면 아래와 같습니다.

 

df.query("open>=2000")
------------------------------------
	nm	open	close
cd			
A060310	3S	2920	2800
A006840	AK홀딩스	2020	2010
A054620	APS홀딩스	3120	3200

 

 

그외의 쿼리 샘플

 

df.query("nm == '3S'")
------------------------------------
	nm	open	close
cd			
A060310	3S	2920	2800


df.query("open > close")
------------------------------------
	nm	open	close
cd			
A060310	3S	2920	2800
A095570	AJ네트웍스	1920	1900
A006840	AK홀딩스	2020	2010


df.query("nm in ['3S', 'AK홀딩스']")
------------------------------------
	nm	open	close
cd			
A060310	3S	2920	2800
A006840	AK홀딩스	2020	2010


df.query("cd == 'A060310'")
------------------------------------
	nm	open	close
cd			
A060310	3S	2920	2800
name = "AJ네트웍스"

# 키워드를 사용하여 변수처럼 사용가능하다 
df.query('nm == @name')
------------------------------------
	nm	open	close
cd			
A095570	AJ네트웍스	1920	1900