1.make_addplot()函数
make_addplot不仅可通过常规变量的可视化,还可可视化一些其他的分析数据,make_addplot可以接受DataFrame、Ndarray、list格式的数据以及**kwargs参数。需要注意的是,传递给make_addplot的数据必须与传递给plot的数据行数相同,**kwargs参数将全部传递到plot方法中。
1.1 计算布林带上中下轨
df['upper'], df['Middler'], df['Lower'] = tb.BBANDS(df['close'], timeperiod=5, nbdevup=2, nbdevdn=2, matype=0) df.fillna(method='bfill',inplace=True) # 用下一个非空值向上填充print(df)
1.2 在主图上绘制附图
add_plot=mpf.make_addplot(df[['upper','Middler','Lower']])mpf.plot(df,type='candle',mav=(5,10,30),volume=True,addplot=add_plot)
1.2.1 将数据分析的结果标记到图像中
在make_addplot方法中使用maker参数,并使用markersize和color设置标记的大小和颜色
这些参数可直接传递给plot方法。
def find_signal(df):# 收盘价上穿布林带上轨做多df.loc[(df['close'].shift(1) <= df['upper'].shift(1)) & (df['close'] > df['upper']), 'signal_long'] = 1# 收盘价下穿布林带中轨平仓df.loc[(df['close'].shift(1) >= df['middler'].shift(1)) & (df['close'] < df['middler']), 'signal_short'] = -1return dfdf=find_signal(df)df.loc[df['signal_short'].notna(),'signal_short']=df['high']add_plot=[mpf.make_addplot(df[['upper','middler','lower']]),mpf.make_addplot(df['signal_short'].values,type='scatter',markersize=20,marker='v',color='g')]mpf.plot(df,type='candle',volume=True,addplot=add_plot)
1.2.2 绘制子图
在make_addplot中使用panel参数绘制子图,panel=0为在主图中绘制,其最多可绘制9张子图。
当同一个副图绘制的图形超过两个小时,可用secondary_y参数,其有三个参数,True、False,auto,默认为auto,
绘制的数据多时最好用secondary_y=True。
add_plot=[mpf.make_addplot(df[['upper','middler','lower']]),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)
1.2.3 调整子图位置
在mpf.plot()方法中使用main_panel这两个关键字,更改主图和成交量子图的位置。
add_plot=[mpf.make_addplot(df[['upper','middler','lower']]),mpf.make_addplot(df['signal_short'].values,type='scatter',markersize=20,marker='v',color='g'),mpf.make_addplot(df['close'].values,panel=1,color='y',secondary_y='auto')]mpf.plot(df,type='candle',volume=True,addplot=add_plot,main_panel=2,volume_panel=0)
1.2.4 设置主图与子图的比例和数量
在mpf.plot()使用num_panels设置图形张数,使用panel_ratios设置各张图之间的比例,panel_ratio=(2,1,1)即图0是2,以此类推。
add_plot=[mpf.make_addplot(df[['upper','middler','lower']]),mpf.make_addplot(df['signal_short'].values,type='scatter',markersize=20,marker='v',color='g'),mpf.make_addplot(df['close'].values,panel=1,color='y',secondary_y='auto')]mpf.plot(df,type='candle',volume=True,addplot=add_plot,main_panel=2,volume_panel=0,num_panels=3,panel_ratios=(2,1,1))
1.2.5 设置图表中线的类型
使用linestyle函数设置线
add_plot=[mpf.make_addplot(df[['upper']],linestyle='dashdot'),mpf.make_addplot(df[['middler']],linestyle='dashdot'),mpf.make_addplot(df[['lower']],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))