Pandasに入れた五分足のデータからローソク足チャートを作りたい。

Question

いくつかのホームページを見て書いてみましたが、うまくいきません

下のコードでpandas dataframeのdfには日付をindexでdatetimeとしてopen, high, low, close, volumeがfloatとして入ることとします。
indexに2020/02/10 13:35の形式で時間が入っていて、どdf['date'] = df.indexで追加したdateカラムが最後のカラムに入りました。この追加したdateカラムはdatetime64[ns]というtypeになっていました。。

実行すると
TypeError: unsupported operand type(s) for -: 'Timestamp' and 'float'となります。

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
from mpl_finance import candlestick_ohlc

    df = get_price_dataframe(db_file_name, code)
    df = f.loc[:, ['open', 'high', 'low', 'close', 'volume']].sort_values('date')

    df['date'] = df.index
    print(df["date"].dtypes)

    fig, ax = plt.subplots(figsize=(10, 5))
    df = df[["date", "open", "high", "low", "close"]]

    candlestick_ohlc(ax, df.values, colorup="green", colordown="red")
    plt.show()

うまくローソク足がプロットできるなら上記コードにはこだわりはありません。というかmpl_financeは古くて、mplfinanceにした方が良いような事を書いてあるホームページもありましたが、mplfinanceではもっと訳が分からない感じにエラーが出て、うまく表示できませんでした……。

Comments

[Daregada]

> mplfinanceではもっと訳が分からない感じにエラーが出て、うまく表示できませんでした……。
具体的なエラーを書いてください。

Answer 1

mpl-financeのcandlestick_ohlcに渡す日時はdatetime64[ns](各要素はTimestamp)ではダメで、質問文のエラーはそれが原因。

Timestamp型のデータを西暦0001-01-01からの経過日数に直す必要がある。それには関数matpltlib.dates.date2numを使うので、import matplotlib.dates as mdatesとでもしておいて、

-df['date'] = df.index
+df['date'] = mdates.date2num(df.index)

のようにすればいい。

と、一応やってみたんだけど面倒すぎるので、mplfinanceに移行することをお薦めする。

おそらく質問者がデータベースから読んでいるのと同型式と思われるCSVファイルを用意し、ローソク足グラフを表示するためのコードがこちら。

import pandas as pd
import mplfinance as mpf
import matplotlib.pyplot as plt

df = pd.read_csv('mplfinance_sample.csv', index_col=0, parse_dates=[0])
mpf.plot(df, type="candle", figratio=(10, 5))
plt.show()

Answer 2

    tmp = df['date'].values.astype('datetime64[D]')  #精度を日単位まで落としてみた
    df['date'] = tmp.astype(float)

これを
df['date'] = df.index
の後に挿入したらエラー消えました。floatに変換しないといけないみたいです。
しかし、x軸は変な実数になってるし、全然五分足にならない……