Python으로 주가 이동평균선 그리기

 

이전까지 구한 값들을 가지고 차트를 그리기 위해 matplotlib 패키지를 설치해줍니다.

pip install matplotlib

 

아래는 전체 코드입니다. 

import pandas_datareader.data as web
import datetime
import matplotlib.pyplot as plt

start = datetime.datetime(2019, 1, 1)
end = datetime.datetime(2019, 10, 29)
ds = web.DataReader("035250.KS", "yahoo", start, end)

ma5 = ds['Adj Close'].rolling(window=5).mean()
ma20 = ds['Adj Close'].rolling(window=20).mean()
ma60 = ds['Adj Close'].rolling(window=60).mean()
ma120 = ds['Adj Close'].rolling(window=120).mean()

ds.insert(len(ds.columns), "MA5", ma5)
ds.insert(len(ds.columns), "MA20", ma20)
ds.insert(len(ds.columns), "MA60", ma60)
ds.insert(len(ds.columns), "MA120", ma120)

plt.plot(ds.index, ds['Adj Close'], label='Adj Close')
plt.plot(ds.index, ds['MA5'], label='MA5')
plt.plot(ds.index, ds['MA20'], label='MA20')
plt.plot(ds.index, ds['MA60'], label='MA60')
plt.plot(ds.index, ds['MA120'], label='MA120')

plt.grid()
plt.show()

요즘 부쩍 관심이 생긴 강원랜드의 이동평균선을 그려보았습니다.