Stock/krx.py

19 lines
600 B
Python
Raw Permalink Normal View History

2022-05-25 03:16:39 +09:00
import pandas as pd
def load_from_krx_server():
code_df = pd.read_html('http://kind.krx.co.kr/corpgeneral/corpList.do?method=download&searchType=13', header=0)[0]
code_df.종목코드 = code_df.종목코드.map('{:06d}'.format)
return code_df
def load_stock_codes():
code_df = pd.read_csv('krx.csv',index_col=0)
code_df.종목코드 = code_df.종목코드.map('{:06d}'.format)
code_df = code_df[['회사명', '종목코드']]
code_df = code_df.rename(columns={'회사명': 'name', '종목코드': 'code'})
stock_codes = code_df["code"]
return stock_codes