[데이터 수집 및 전처리] [Python/파이썬]네이버증권API 활용 - 회사명, 종목코드 받아오기

*Python 3.9.13을 사용했습니다.

한국거래소에서는 국내 상장사(코스피/코스닥)에 대한 정보를 아래 링크의 엑셀 파일을 통해 제공하고 있습니다.

이 파일에서 "회사명, 종목코드"만 수집하여 추후 네이버API 사용 시 활용하고자 합니다.

우선, Pandas를 설치한 후 Pandas 모듈의 read_html을 사용하여 한국거래소 정보 파일을 읽어 옵니다.

import pandas as pd
#한국거래소에서 종목 코드 받아옵니다. [0]은 헤더를 첫번째 행으로 지정하기 위해 사용 했습니다.
code_df = pd.read_html('http://kind.krx.co.kr/corpgeneral/corpList.do?method=download&searchType=13')[0]

위 코드의 마지막 줄에 print (code_df)를 실행하면 아래와 같이 엑셀의 정보가 code_df변수에 저장된 것을 확인할 수 있습니다.

다만, 앞으로 있을 네이버증권API 사용 시 종목코드가 6자리여야 하기에 아래와 같이 종목코드의 포맷을 변경해줍니다.

# 종목코드가 6자리이기 때문에 6자리를 맞춰주기 위해 설정해줍니다.
code_df.종목코드 = code_df.종목코드.map('{:06d}'.format)

위 코드의 마지막 줄에 print(code_df)를 실행하면 종목코드의 포맷이 변경된 것을 확인할 수 있습니다.

경축! 아무것도 안하여 에스천사게임즈가 새로운 모습으로 재오픈 하였습니다.
어린이용이며, 설치가 필요없는 브라우저 게임입니다.
https://s1004games.com

이제 우리가 필요한 것은 "회사명, 종목코드" 이기에 다른 컬럼을 읽기에서 제외 시킵니다. 또한, 리스트로 변경하여 앞으로 있을 네이버증권API 사용에 활용할 예정입니다.

#우리에게 필요한 회사명과 종목코드만 남깁니다.
code_df = code_df[['회사명', '종목코드']]
#리스트 형식으로 변경하면 네이버증권API 사용을 위한 준비가 끝납니다.
stock_list = code_df.values.tolist()

stock_list를 출력해보면 아래와 같이 리스트로 잘 변경된 것을 볼 수 있습니다.

전체 코드는 아래와 같습니다. 이제 다음 시간엔 stock_list 변수를 활용할 예정 입니다.

import pandas as pd
#한국거래소에서 종목 코드 받아옵니다. [0]은 헤더를 첫번째 행으로 지정하기 위해 사용 했습니다.
code_df = pd.read_html('http://kind.krx.co.kr/corpgeneral/corpList.do?method=download&searchType=13')[0]
#종목코드가 6자리이기 때문에 6자리를 맞춰주기 위해 설정해줍니다.
code_df.종목코드 = code_df.종목코드.map('{:06d}'.format)
#우리에게 필요한 회사명과 종목코드만 남깁니다.
code_df = code_df[['회사명', '종목코드']]
#리스트 형식으로 변경하면 네이버증권API 사용을 위한 준비가 끝납니다.
stock_list = code_df.values.tolist()

출처: https://marketward.tistory.com/2 [코드제이:티스토리]