量化交易 - 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)