공부/퀀트
파이썬을 이용해 급등 종목 추출하기
민수
2022. 1. 7. 16:28
In [8]:
from IPython.core.display import display, HTML
display(HTML("<style>.container { width:90% !important; }</style>"))
In [1]:
import pandas as pd
import numpy as np
import FinanceDataReader as fdr
import matplotlib.pyplot as plt
from tqdm import tqdm
from datetime import datetime
from dateutil.relativedelta import relativedelta
FinanceDataReader를 이용해 종목 데이터 전체를 가져오기¶
- KRX : KRX 종목 전체
- KOSPI : KOSPI 종목
- KOSDAQ : KOSDAQ 종목
- KONEX : KONEX 종목
In [2]:
# 한국거래소 상장종목 전체
df_kosdaq = fdr.StockListing('KOSDAQ')
print(df_kosdaq.shape)
df_kosdaq.head()
(1552, 10)
Out[2]:
| Symbol | Market | Name | Sector | Industry | ListingDate | SettleMonth | Representative | HomePage | Region | |
|---|---|---|---|---|---|---|---|---|---|---|
| 0 | 060310 | KOSDAQ | 3S | 전자부품 제조업 | 반도체 웨이퍼 캐리어 | 2002-04-23 | 03월 | 박종익, 김세완 (각자 대표이사) | http://www.3sref.com | 서울특별시 |
| 3 | 054620 | KOSDAQ | APS홀딩스 | 기타 금융업 | 인터넷 트래픽 솔루션 | 2001-12-04 | 12월 | 정기로 | http://www.apsholdings.co.kr | 경기도 |
| 4 | 265520 | KOSDAQ | AP시스템 | 특수 목적용 기계 제조업 | 디스플레이 제조 장비 | 2017-04-07 | 12월 | 김영주 | http://www.apsystems.co.kr | 경기도 |
| 5 | 211270 | KOSDAQ | AP위성 | 통신 및 방송 장비 제조업 | 위성통신 단말기 | 2016-03-04 | 12월 | 류장수 | http://www.apsi.co.kr | 서울특별시 |
| 50 | 032790 | KOSDAQ | BNGT | 기계장비 및 관련 물품 도매업 | Bio 이종장기 사업, ICT 프린터 현상기 | 1997-06-26 | 12월 | 조상환 | http://www.mgenplus.com | 서울특별시 |
In [3]:
# 오늘 날짜를 불러옴
today = str(pd.to_datetime(datetime.now()))[:10]
today
Out[3]:
'2022-01-07'In [4]:
# 오늘 날짜의 모든 주식을 담음
all_df = pd.DataFrame()
for i in tqdm(df_kosdaq.Symbol):
df = fdr.DataReader(i,today,today) # fdr을 통해 특정 기업의 주가를 가져옴(오늘)
df['Symbol'] = i # 종목 코드를 포함시켜서 저장
all_df = all_df.append(df)
100%|██████████████████████████████████████████████████████████████████████████████| 1552/1552 [02:00<00:00, 12.89it/s]
In [5]:
# 원하는 조건식을 통해 종목 검출
filtered_df = all_df[all_df['Change']>0.2] # 어제에 비해 변화량이 0.2(20퍼) 이상인 종목을 검출
filtered_df
Out[5]:
| Open | High | Low | Close | Volume | Change | Symbol | |
|---|---|---|---|---|---|---|---|
| 2022-01-07 | 10400.0 | 13300.0 | 10350.0 | 13300.0 | 2661154.0 | 0.297561 | 059120 |
| 2022-01-07 | 4680.0 | 5590.0 | 4680.0 | 5590.0 | 2726750.0 | 0.298490 | 196700 |
| 2022-01-07 | 5570.0 | 7000.0 | 5560.0 | 6870.0 | 31668958.0 | 0.244565 | 056080 |
| 2022-01-07 | 2825.0 | 3615.0 | 2650.0 | 3440.0 | 63678881.0 | 0.219858 | 065440 |
| 2022-01-07 | 4775.0 | 6200.0 | 4700.0 | 6000.0 | 11053556.0 | 0.252610 | 114630 |
| 2022-01-07 | 20500.0 | 25450.0 | 20100.0 | 25450.0 | 1041623.0 | 0.298469 | 03481K |
In [6]:
# 종목 코드를 활용하여 기업 코드를 매칭시킴
df_kosdaq = df_kosdaq[['Symbol','Name']] # 코스닥 종목데이터에서 종목코드,기업명만 남기고 제거
pd.merge(filtered_df,df_kosdaq) # 조건식으로 검출한 종목과 병합하여 기업명 생성
Out[6]:
| Open | High | Low | Close | Volume | Change | Symbol | Name | |
|---|---|---|---|---|---|---|---|---|
| 0 | 10400.0 | 13300.0 | 10350.0 | 13300.0 | 2661154.0 | 0.297561 | 059120 | 아진엑스텍 |
| 1 | 4680.0 | 5590.0 | 4680.0 | 5590.0 | 2726750.0 | 0.298490 | 196700 | 웹스 |
| 2 | 5570.0 | 7000.0 | 5560.0 | 6870.0 | 31668958.0 | 0.244565 | 056080 | 유진로봇 |
| 3 | 2825.0 | 3615.0 | 2650.0 | 3440.0 | 63678881.0 | 0.219858 | 065440 | 이루온 |
| 4 | 4775.0 | 6200.0 | 4700.0 | 6000.0 | 11053556.0 | 0.252610 | 114630 | 폴라리스우노 |
| 5 | 20500.0 | 25450.0 | 20100.0 | 25450.0 | 1041623.0 | 0.298469 | 03481K | 해성산업1우 |