본문 바로가기

포트폴리오/api

mpl_finance 에서 candlestick_ohlc 사용하기.

간단한 예제코드. 삼성전자 캔들스틱 확인하기. "파이썬으로 배우는 알고리즘 트레이딩" 서적과 mpl_finance 깃허브를 참고하였습니다.
	
import datetime
import matplotlib.dates as mdates
import matplotlib.pyplot as plt
import pandas_datareader as web
from mpl_finance import candlestick_ohlc

start = datetime.datetime(2018, 12, 1)
end = datetime.datetime(2018, 12, 31)
# 코스닥은 "코드.KQ" 사용
samsung = web.DataReader("005930.KS", "yahoo", start, end)

fig = plt.figure(figsize=(12, 8))
ax = fig.add_subplot(111)

# "from mpl_finance 의 candlestick_ohlc()" 입니다.
candlestick_ohlc(ax,zip(mdates.date2num(samsung.index.to_pydatetime()),samsung['Open'],samsung['High'],samsung['Low'],samsung['Close']),width=0.5,colorup="r",colordown="b")
plt.show()
	


결과는 아래와 같습니다.