2018年10月1日月曜日

jupyterとiexfinanceで米国株価データを取得して、ローソク足グラフを描画する


以下のコードでiexfinanceを使用して米国株価データを取得して、ローソク足グラフを描画する事ができます。

〇出力グラフ


〇コード
import matplotlib.pyplot as plt
from mpl_finance import candlestick_ohlc
import matplotlib.dates as mdates
import numpy as np
from iexfinance import get_historical_data
from datetime import datetime

start = datetime(2018, 6, 1)
end = datetime(2018, 8, 31)

df = get_historical_data("AAPL", start=start, end=end, output_format='pandas')
#print(df)
fig = plt.figure()
ax = plt.subplot()

numrows = len(df)
ohlc = np.vstack((range(numrows), df.values.T)).T
candlestick_ohlc(ax, ohlc, width=0.8, colorup='g', colordown='r')
offset_mon = (5-datetime.strptime(df.index[0], '%Y-%m-%d').weekday())%5

plt.xticks(range(offset_mon,numrows,5), [x for x in df.index][offset_mon::5])
ax.grid(True)
ax.set_xlim(-1, numrows)
fig.autofmt_xdate()

〇必要パッケージのインストール
pip install mpl_finance
pip install iexfinance

○関連情報
JupyterとdatapackageでS&P500株価指数データを取得する

0 件のコメント:

コメントを投稿