pythonのmatplotlib（とちょこっとpandas）でローソク足のチャートを書く
pythonで株やFXなんかで使うローソク足チャートを書きたかったのですが、ちょっと苦戦したのでメモ。 完成品はこんな感じです。
ソースコードはこんな感じ。
import pandas import matplotlib.pyplot as plt from matplotlib.finance import candlestick_ohlc dat = pandas.read_csv('usdjpy.csv', parse_dates=['日付']) # ファイルの読み込み。 dat = dat[-50:] # データが多すぎるので減らす。 dates = dat['日付'] # あとでつかう。 tmp = dat['日付'].values.astype('datetime64[D]') # ナノ秒精度とか無意味なので、精度を日単位まで落とす。 dat['日付'] = tmp.astype(float) # Datetime64形式だと使えないので、floatに変換。 plt.xticks( # 横軸の値と表示の対応の設定。[::5]はラベルを1週間ごとにするために使っている。 dat['日付'][::5], [x.strftime('%Y-%m-%d') for x in dates][::5] ) plt.grid() ax = plt.subplot() candlestick_ohlc( # グラフを作る。 ax, dat.values, # 入力データ。左から順に、始値、高値、安値、終値にする。その後にデータが続いてても良いらしい。 width=0.7, # 棒の横幅。今回は日単位の精度に落としてあるので、0.7日分の幅になる。 colorup='skyblue', colordown='black' ) plt.show()
わりと簡単。
データファイルはマネースクウェア・ジャパンが配布しているドル円のcsvファイルを使う前提で書いています。 まあ、始値高値安値終値の順番が正しければなんでも良い。
candlestick_ochlなんて関数もあって、こっちは始値終値高値安値の順番になります。お好みで。
横軸の表示を設定する方法は色々あるようなのですが、もっとも単純っぽいこの方法にしました。 あまりにもデータ数が多いときには問題が出るかもしれませんが、そうでなければこれで良いはず。
お手軽にわりと綺麗なグラフが出て素敵なんですが、細かい調整はどうも出来ないようです。 箱に枠線を付けるとか、そういうのは無理っぽい。もしくはすごく面倒臭そう。
参考：