머신러닝 파이썬
캔들스틱 차트로 이동평균 주가 그래프 그리기: 파이썬
fast learner
2023. 3. 27. 15:53
import pandas as pd
goog = pd.read_csv('../data/GOOG.csv')
goog['Date'] = pd.to_datetime(goog['Date'])
goog.set_index('Date', inplace=True)
goog = goog[['Open', 'High', 'Low', 'Close', 'Volume']] # index 컬럼인 Date는 필요x
import mplfinance as mpf
goog['20_MA'] = goog['Close'].rolling(window=20).mean()
goog['50_MA'] = goog['Close'].rolling(window=50).mean()
ap_mav = [mpf.make_addplot(goog['20_MA']), mpf.make_addplot(goog['50_MA'], color='red')]
fig, axs = mpf.plot(goog, type='candle', style='charles', title='Stock Price', ylabel='Price', volume=True,
addplot=ap_mav)
axs[0].legend(['20_MA', '50_MA'])

728x90
반응형