量化交易 - matplotlib画candle图

需要mplfinance包

pip install mplfinance --upgrade

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

plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False

# mav=(2,5,7)颜色一般不动
kwargs = dict(type='candle',
              mav=(2, 5, 7),
              volume=True,
              title="股票价格",
              ylabel='价格',
              ylabel_lower='volume',
              figratio=(15, 9),
              figscale=1.3,
              linecolor='g')

mc = mpf.make_marketcolors(up='red',
                           down='green',
                           edge='i',
                           wick='i',
                           volume='in',
                           inherit=True)

mystyle = mpf.make_mpf_style(gridaxis='both',
                             gridstyle='-.',
                             y_on_right=False,
                             marketcolors=mc,
                             base_mpf_style='yahoo',
                             rc={'font.family': 'SimHei'})

f = df[['交易日期', '开盘价', '最高价', '最低价', '收盘价', '成交额']]
f.rename(columns={
    '交易日期': 'Date',
    '开盘价': 'Open',
    '最高价': 'High',
    '最低价': 'Low',
    '收盘价': 'Close',
    '成交额': 'Volume'
},
         inplace=True)
f.set_index(['Date'], inplace=True)
mpf.plot(f, **kwargs, style=mystyle)