mplfinance 使用make_addplot做复杂股票走势图

1.代码

import talib as tb
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 # 解决保存图像符号“-”显示问题

def find_signal(df):
    # 收盘价上穿布林带上轨做多,增加signal_long 列
    df.loc[(df['close'].shift(1) <= df['upper'].shift(1)) & (df['close'] > df['upper']), 'signal_long'] = 1
    # 收盘价下穿布林带中轨平仓,增加signal_short 列
    df.loc[(df['close'].shift(1) >= df['middler'].shift(1)) & (df['close'] < df['middler']), 'signal_short'] = -1

    return df

#df = df.loc['2020-06-01':'2020-12-31']
#df.rename(columns={'vol': 'volume'}, inplace=True)

df['upper'], df['middler'], df['lower'] = tb.BBANDS(df['close'], timeperiod=5, nbdevup=2, nbdevdn=2, matype=0)
df.fillna(method='bfill',inplace=True) # 用下一个非空值向上填充

df=find_signal(df)

df.loc[df['signal_short'].notna(),'signal_short']=df['high']

my_color=mpf.make_marketcolors(up='red',down='green',edge='black',wick='i',volume={'up':'red','down':'green'},ohlc='black',inherit=False)
my_style=mpf.make_mpf_style(base_mpf_style='sas',marketcolors=my_color,gridaxis='both',gridstyle='-.',y_on_right=False,rc={'font.family':'SimHei'})

add_plot=[mpf.make_addplot(df[['upper']].values,linestyle='dashdot'),
          mpf.make_addplot(df[['middler']].values,linestyle='dashdot'),
          mpf.make_addplot(df[['lower']].values,linestyle='dashdot'),
          mpf.make_addplot(df[['signal_short']].values,type='scatter',markersize=20,marker='v',color='g'),
          mpf.make_addplot(df[['close']].values,panel=2,color='y',secondary_y='auto')]

mpf.plot(df,type='candle',volume=True,addplot=add_plot,main_panel=0,volume_panel=1,
         num_panels=3,panel_ratios=(2,1,1),style=my_style, title='布林线',ylabel='价格',ylabel_lower='成交量',figratio=(9,6),figscale=1.2,show_nontrading=False)

2.图示效果

3.说明

1.make_mpf_style()函数
make_mpf_style(base_mpf_style,base_mpl_style,marketcolors,mavcolors,facecolor,edgecolor,figcolor,gridcolor,gridaxis,gridstyle,y_on_right,rc)

函数部分参数如下：

• 1）.base_mpf_style设置需要继承的系统样式；
• 2）.base_mpl_style设置同时使用matplotlib中的式样seaborn；
• 3）.marketcolor设置K线的颜色，可以使用make_marketcolors()函数进行定义；
• 4）.mavcolors设置均线的颜色，必须使用列表传参；
• 5）.facecolor设置前景色；
• 6）.edgecolor设置边缘线颜色；
• 7）.figcolor设置图像外周围填充色；
• 8）.gridcolor设置网格线颜色；
• 9）.gridaxis设置网格线的位置，gridaxis=‘both’/‘horizontal’/‘vertical’；
• 10）.gridstyle设置网格线线型，gridstyle=‘solid’/‘dashed’/‘dashdot’/‘dotted’；
• 11）.y_on_right设置y轴位置是否在右,y_on_right=True设为右边；
• 12）.rc使用rcParams的dict设置格式；

2.关于make_mpf_style()函数中的marketcolors参数
make_marketcolors(up,down,edge,wick,volume,ohlc,inherit)
函数部分参数如下：

• 1).up设置阳线柱填充颜色；
• 2).down设置阴线线柱填充颜色；
• 3).edge设置蜡烛线边缘颜色，edge='i’代表继承K线主体颜色；
• 4).wick设置蜡烛线上下影线的颜色，wick='i’代表继承K线主体的颜色；
• 5).volume设置成交量的颜色，volume='i’代表继承K线主体颜色；
• 6).ohlc设置均线颜色，代表继承K线主体的颜色；
• 7).inherit设置是否继承，如果设置了继承inherit=True,那么edge/wick/volume/ohlc 四个参数即便设置了颜色也会无效，其中edge/wick/volume/ohlc四个参数除了设置’i’ 自动继承up和down的颜色外，也可以使用dict模式定义。

参考：
https://blog.csdn.net/malishizu222/article/details/124484598
https://blog.csdn.net/qq_39065491/article/details/130010354

4. 问题

make_addplot函数的一些主要参数说明：

数据参数：

• Ndarray、list：make_addplot可以接受这些格式的数据作为输入。这意味着你可以传递一个pandas DataFrame、numpy Ndarray或一个Python list来包含你想要在主图上展示的额外数据。
• 特殊参数：
  **kwargs：这是一个可变关键字参数，意味着你可以传递任何额外的关键字参数给make_addplot。这些参数随后将被传递到绘图方法（如plot）中。
• 注意事项：
  传递给make_addplot的数据参数行数必须与将来传递给plot方法的数据行数相同。这是因为make_addplot是为了在主价格图下方同步地添加额外的图表或数据。
• 标记参数：
  marker：用于设置标记的形状，例如^表示正三角形，v表示倒三角形。
  markersize：用于设置标记点的大小。
  color：用于设置标记点的颜色。请注意，这个颜色设置是针对标记点的，而不是线条的颜色。
  布局参数：
• panel：当设置为"lower"时，可以让线条显示在附图的成交量上方。

make_addplot()必须使用Ndarray，否则报错

print(type(df[['upper']]))
print(type(df['upper']))
print(type(df['close'].values))

结果是：

<class 'pandas.core.frame.DataFrame'>
<class 'pandas.core.series.Series'>
<class 'numpy.ndarray'>

必须用 df[‘close’].values ！！！
不是ndarray类型，直接把报错，折腾了半天才找到原因！！！

ValueError: Multi-dimensional indexing (e.g. obj[:, None]) is no longer supported. Convert to a numpy array before indexing instead.