프로그래밍 노트

[파이썬] 웹에서 상장된 회사 명과 주식 코드가져오기 본문

파이썬

[파이썬] 웹에서 상장된 회사 명과 주식 코드가져오기

떡잎 2020. 1. 24. 21:05

 

 

 

 

In [1]:
import pandas as pd

dfstockcode = pd.read_html('http://kind.krx.co.kr/corpgeneral/corpList.do?method=download', header=0)[0]
In [2]:
dfstockcode.columns
Out[2]:
Index(['회사명', '종목코드', '업종', '주요제품', '상장일', '결산월', '대표자명', '홈페이지', '지역'], dtype='object')
In [3]:
# 얻고자하는 DataFrame의 컬럼명을 쓰면 
# 시리즈형으로 컬럼에 있는 값을 반환
dfstockcode.종목코드.head()
Out[3]:
0      5560
1    101380
2    114140
3     64900
4     28780
Name: 종목코드, dtype: int64
In [4]:
type(dfstockcode.종목코드)
Out[4]:
pandas.core.series.Series
In [5]:
# 종목코드에 값을 6자리 문자열로 설정 
#    종목코드의 숫자값을 6자리 문자열로 변환
#    채우는 자리는 0으로 채우기
dfstockcode.종목코드 = dfstockcode.종목코드.map('{:06d}'.format)
In [6]:
# 필요한 컬럼만으로 DataFrame을 설정
dfstockcode = dfstockcode[['회사명', '종목코드']]
In [7]:
dfstockcode.종목코드.head()
Out[7]:
0    005560
1    101380
2    114140
3    064900
4    028780
Name: 종목코드, dtype: object
In [8]:
# 컬럼명을 한글에서 영어로 변경 
dfstockcode = dfstockcode.rename(columns={'회사명':'name', '종목코드':'code'})
In [9]:
dfstockcode.head()
Out[9]:
  name code
0 JS전선 005560
1 거북선2호 101380
2 거북선6호 114140
3 교보메리츠 064900
4 국제관광공사 028780
 
 
 
웹에서 우리나라에서 상장된 모든 회사명과 주식코드명을 DataFrame 형으로 가져오기
 
import pandas as pd

dfstockcode = pd.read_html('http://kind.krx.co.kr/corpgeneral/corpList.do?method=download', header=0)[0]

# 종목코드에 값을 6자리 문자열로 설정 
#    종목코드의 숫자값을 6자리 문자열로 변환
#    채우는 자리는 0으로 채우기
dfstockcode.종목코드 = dfstockcode.종목코드.map('{:06d}'.format)

# 필요한 컬럼만으로 DataFrame을 설정
dfstockcode = dfstockcode[['회사명', '종목코드']]

# 컬럼명을 한글에서 영어로 변경 
dfstockcode = dfstockcode.rename(columns={'회사명':'name', '종목코드':'code'})

역시 파이썬 정말 놀랍다.

정말 몇 줄안되는 코드로 이런 것을 할 수 있다나....

Comments