이하영

기업목록 받아오기

import pandas as pd
class itemList():
def __init__(self):
self.get_kospi()
print("코스피 종목 수 : ",len(self.code_df_kospi))
print(self.code_df_kospi)
self.get_kosdaq()
print("코스닥 종목 수 : ",len(self.code_df_kosdaq))
print(self.code_df_kosdaq)
def get_download_stock(self,market_type=None):
stock_type={'kospi':'stockMkt','kosdaq':'kosdaqMkt'}
market_type=stock_type[market_type]
# 한국거래소 - 상장공시시스템 링크를 이용
# searchType=13 : 상장기업을 대상으로 조회
download_link = 'http://kind.krx.co.kr/corpgeneral/corpList.do?method=download&searchType=13'
download_link = download_link + '&marketType=' + market_type
df=pd.read_html(download_link,header=0)[0]
return df
def get_kospi(self):
self.code_df_kospi=self.get_download_stock('kospi')
self.code_df_kospi.종목코드=self.code_df_kospi.종목코드.map('{:06d}'.format) # 종목코드는 6자리
self.code_df_kospi=self.code_df_kospi[['회사명','종목코드']]
self.code_df_kospi = self.code_df_kospi.rename(columns={'회사명': 'code_name', '종목코드': 'code'}) # 한글 컬럼명을 영어로 변경
return self.code_df_kospi
def get_kosdaq(self):
self.code_df_kosdaq=self.get_download_stock('kosdaq')
self.code_df_kosdaq.종목코드=self.code_df_kosdaq.종목코드.map('{:06d}'.format)
self.code_df_kosdaq=self.code_df_kosdaq[['회사명','종목코드']]
self.code_df_kosdaq = self.code_df_kosdaq.rename(columns={'회사명': 'code_name', '종목코드': 'code'})
return self.code_df_kosdaq
if __name__== "__main__":
itemList()
\ No newline at end of file