주가 데이터 수집 및 차트 그리기

주식 데이터는 오픈 API를 통해 많이 공개돼 있습니다. 그래서 수집하고 분석하기에 용이합니다. 오픈 API에서 제공하는 데이터를 활용해 더 쉽게 사용할 수 있게 도와주는 패키지도 있습니다. FinanceDataReader도 그중 하나인데요. 오늘은 이 패키지를 이용해 주가 데이터를 수집하고, 차트를 그려보겠습니다.

 

 

1. FinanceDataReader 설치

FinanceDataReader는 pip 명령을 이용해서 쉽게 설치할 수 있습니다. 설치를 위해 입력하는 패키지명과 import 할 때 사용하는 이름이 다르다는 것만 염두에 두면 됩니다.

 

pip install finance_datareader

 

 

2. 주가 데이터 수집하기

DataReader 함수에 종목코드와 기간을 입력하면 주가 데이터를 수집할 수 있습니다. 삼성전자,LG전자 주가 데이터를 2015년부터 2020년 7월 31일까지 수집해 봤습니다.

import FinanceDataReader as fdr

ss=fdr.DataReader("005930","2015","2020-07-31")
lg=fdr.DataReader("066570","2015","2020-07-31")

 

head함수를 이용해서 앞의 일부 행을 확인해 봤습니다.

ss.head()

 

주가데이터-수집

 

 

3. 차트 그리기

주가 차트는 Candle차트라고 부르는데요. 캔들 차트를 그릴 수 있는 패키지는 여러가지가 있습니다. 여기서는 plotly 패키지를 이용해서 주가 차트를 그려 봤습니다.

 

import plotly.graph_objects as go

go.Figure(data=go.Candlestick(x=ss.index,
          open=ss['Open'],
          high=ss['High'],
          low=ss['Low'],
          close=ss['Close']))

 

삼성전자-주식-차트

 

오늘은 이렇게 주가 데이터 수집 및 차트 그려 봤습니다. 다음에는 이동평균선을 그리고, 차트를 그려보는 작업을 해 보겠습니다.