19 lines
600 B
Python
19 lines
600 B
Python
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
|
|
|
|
|