2019年5月9日木曜日

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

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

〇出力グラフ


〇コード
import matplotlib.pyplot as plt
from mpl_finance import candlestick_ohlc
import matplotlib.dates as mdates
import numpy as np
import iexfinance as iexf
from iexfinance.stocks import get_historical_data
from datetime import datetime
print("iexfinance:" + iexf.__version__)

start = datetime(2019, 1, 1)
end = datetime(2019, 5, 5)
df = get_historical_data("AAPL", start=start, end=end, output_format='pandas')
df = df.reset_index()
df['datenum'] = df['date'].apply(lambda date: mdates.date2num(date.to_pydatetime()))
df = df.set_index('datenum').drop(columns="date")

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-mdates.num2date(df.index[0]).weekday())%5
plt.xticks(range(offset_mon,numrows,5), [mdates.num2date(x).strftime("%Y-%m-%d") for x in df.index][offset_mon::5])
ax.grid(True)
ax.set_xlim(-1, numrows)
fig.autofmt_xdate()

〇上記のソースコードは以下の記事のDockerコンテナで実行することができます。
DockerでJupyter Lab、iexfinanceとipython-sql、PostgreSQL11がインストールされたコンテナを作成する

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

0 件のコメント:

コメントを投稿