"이평선(MA) 차트 그리기" 파이썬으로 배우는 알고리즘 투자 스터디자료

"이평선(MA) 차트 그리기" 스터디자료6. 파이썬으로 배우는 알고리즘 투자 

import pandas as pd

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

# Get GS Data from Yahoo
gs = web.DataReader("078930.KS", "yahoo", "2015-01-01", "2018-05-06")
new_gs = gs[gs['Volume']!=0]

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

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

# Plot

# 범례를 만드는 가장 쉬운 방법은 그래프를 그릴 때 그래프에 label 인자를 사용하는 것입니다.
plt.plot(new_gs.index, new_gs['Adj Close'], label='Adj Close')
plt.plot(new_gs.index, new_gs['MA5'], label='MA5')
plt.plot(new_gs.index, new_gs['MA20'], label='MA20')
plt.plot(new_gs.index, new_gs['MA60'], label='MA60')
plt.plot(new_gs.index, new_gs['MA120'], label='MA120')


# 범례를 표시하기 위해 legend 함수를 호출합니다.
# 이때 loc 인자를 통해 범례 표시 위치를 지정할 수 있습니다.
# 범례가 적절한 위치에 자동으로 출력되게 하려면 loc='best' 옵션을 사용하면 됩니다.
# 또한 그래프의 값을 좀 더 편리하게 확인하기 위한 격자(grid)를 표시하려면 grid 함수를 호출하면 됩니다.

plt.legend(loc="best")

plt.grid()

plt.show()

스터디 자료의 출처는

파이썬으로 배우는 알고리즘 트레이딩(조대표 외 1명)입니다.

아래 책을 터치하시면 구매페이지로 연결됩니다. 

스터디 진행하며 사용한 비영리 목적의 참고자료로

크레이티브 커먼즈에 따라 사용하며,

저작권자의 삭제 요청이 있을시 삭제하겠습니다.

공감과 댓글은 글쓴이에게 큰 힘이 됩니다. 

마음에 드셨으면 공감과 댓글부탁드립니다.