Python的新mplfinance模块入门介绍

  一、make_addplot的基本用法

  在金融数据分析中,我们要通过数据可视化展示的不仅是’Open’, ‘High’, ‘Low’, 'Close’和最常见‘ma’,还有一些其他分析数据,那么就要用到make_addplot()方法了,make_addplot可以接受一个pandas、numpy、array以及list格式的数据(tuple不可以),和**kwargs参数;需要注意的是:传递给make_addplot的数据参数必须与将来画图传递给plot的数据参数行数相同,**kwargs参数将全部传递到polt方法中。

  下面准备了一组数据,但是column有点长,不能全部输出,这里分布输出了数据的columns和一部分数据:

  Index(['Open', 'High', 'Low', 'Close', 'Volume', 'UpperB', 'LowerB', 'PercentB'],

  dtype='object')

  Open High ... LowerB PercentB

  Date ...

  2011-07-01 132.089996 134.100006 ... 125.316073 1.219057

  2011-07-05 133.779999 134.080002 ... 124.912703 1.066618

  2011-07-06 133.490005 134.139999 ... 124.627085 0.992467

  2011-07-07 135.160004 135.699997 ... 124.284986 1.026222

  2011-07-08 133.830002 135.360001 ... 124.140890 0.885916

  ... ... ... ... ... ...

  2012-06-25 132.050003 132.100006 ... 128.120436 0.381896

  2012-06-26 131.699997 132.380005 ... 128.084306 0.470585

  2012-06-27 132.419998 133.429993 ... 128.140042 0.605441

  2012-06-28 132.289993 132.990005 ... 128.219241 0.551922

  2012-06-29 135.199997 136.270004 ... 128.792993 0.921670

  [252 rows x 8 columns]

  add_plot = mplfinance.make_addplot(data['LowerB'])

  mplfinance.plot(data, addplot=add_plot)

  plt.show() # 显示

  得出结果

  如果要给图表添加多个数据绘制,直接用列表传入make_addplot即可,例如:

  add_plot = mplfinance.make_addplot(data[['UpperB', 'LowerB']])

  mplfinance.plot(data, addplot=add_plot)

  plt.show() # 显示

  得出结果

  把数据分析的结果标记到图像中

  数据分析和可视化的目的是有分析结果,直接把结果展示的图表上将更加直观。先做个简单的数据分析,并把分析结果赋值到两个列表中,然后在绘图时标记到图表中,因为这里是标记,并非连续的线

  ,此时可以在make_addplot方法中使用marker,以及markersize和color设置标记的大小和颜色,这些参数都是直接传递给plot方法的。

  def data_analyze(self, data: pandas.DataFrame):

  """

  简单的数据分析,并把返回数据分析结果列表,分析的逻辑不重要,主要看如何绘制到图形中。

  :param data:

  :return:

  """

  if data.shape[0] == 0:

  data = self.data

  s_list = []

  b_list = []

  b=-1

  for i, v in data['High'].iteritems():

  if v > data['UpperB'][i] and (b == -1 or b == 1):

  b_list.append(data['Low'][i])

  b = 0

  else:

  b_list.append(numpy.nan) # 这里添加nan的目的是,对齐主图的k线数量

  if data['Low'][i] < data['LowerB'][i] and (b == -1 or b == 0):

  s_list.append(v)

  b = 1

  else:

  s_list.append(numpy.nan)

  return b_list, s_list

  b_list, s_list = self.data_analyze(data)

  add_plot = [

  mpf.make_addplot(b_list, scatter=True, markersize=200, marker='^', color='y'),

  mpf.make_addplot(s_list, scatter=True, markersize=200, marker='v', color='r'),

  mpf.make_addplot(data[['UpperB', 'LowerB']])]

  mpf.plot(data, type='candle', addplot=add_plot, volume=True)

  plt.show() # 显示

  得出结果

  在副图中绘制

  plot绘图时是可以直接通过参数添加成交量副图的,如果想在副图中绘制数据可以在make_addplot中使用参数panel,直接上代码:

  def data_analyze(self, data: pandas.DataFrame):

  """

  简单的数据分析,并把返回数据分析结果列表,分析的逻辑不重要,主要看如何绘制到图形中。

  :param data:

  :return:

  """

  if data.shape[0] == 0:

  data = self.data

  s_list = []

  b_list = []

  b=-1

  for i, v in data['High'].iteritems():

  if v > data['UpperB'][i] and (b == -1 or b == 1):

  b_list.append(data['Low'][i])

  b = 0

  else:

  b_list.append(numpy.nan) # 这里添加nan的目的是,对齐主图的k线数量

  if data['Low'][i] < data['LowerB'][i] and (b == -1 or b == 0):

  s_list.append(v)

  b = 1

  else:

  s_list.append(numpy.nan)

  return b_list, s_list

  b_list, s_list = self.data_analyze(data)

  add_plot = [

  mpf.make_addplot(b_list, scatter=True, markersize=200, marker='^', color='y'),

  mpf.make_addplot(s_list, scatter=True, markersize=200, marker='v', color='r'),

  mpf.make_addplot(data[['UpperB', 'LowerB']]),

  mpf.make_addplot(data['PercentB'], panel='lower', color='g', secondary_y='auto'),]

  mpf.plot(data, type='candle', addplot=add_plot, volume=True)

  plt.show() # 显示

  得出结果

  这里注意的是副图的左右两侧都有Y轴标,而且数据不同,因为mplfinance为完整展示图像,自动分成了两个坐标轴,如果你绘制的数据较多,这里有可能会报错,如果出现这种情况,在使用plot绘制多组数据时需要使用参数secondary_y,有三个参数选项True, False, 和auto,默认是auto,因为绘制数据多出现报错时只能设置secondary_y=true(使用Y轴标)或/False(不使用)来覆盖,如果数据差异很大,设为False可能会导致看不到(实际上还是绘制了,只是在最顶上或最底下一条线看不清而已)。

  修改主图Y轴刻度位置和设置线形

  在绘图plot函数中使用关键字参数style='sas’可以把主图Y左边放到右边,默认是‘default’,

  设置线型是参数linestyle,直接上代码吧,都是常用的参数:

  def data_analyze(self, data: pandas.DataFrame):

  """

  简单的数据分析,并把返回数据分析结果列表,分析的逻辑不重要,主要看如何绘制到图形中。

  :param data:

  :return:

  """

  if data.shape[0] == 0:

  data = self.data

  s_list = []

  b_list = []

  b=-1

  for i, v in data['High'].iteritems():

  if v > data['UpperB'][i] and (b == -1 or b == 1):

  b_list.append(data['Low'][i])

  b = 0

  else:

  b_list.append(numpy.nan) # 这里添加nan的目的是,对齐主图的k线数量

  if data['Low'][i] < data['LowerB'][i] and (b == -1 or b == 0):

  s_list.append(v)

  b = 1

  else:

  s_list.append(numpy.nan)

  return b_list, s_list

  b_list, s_list = self.data_analyze(data)

  add_plot = [郑州人流医院 http://www.0371zzkd.com/

  mpf.make_addplot(b_list, scatter=True, markersize=200, marker='*', color='y'),

  mpf.make_addplot(s_list, scatter=True, markersize=200, marker='s', color='r'),

  mpf.make_addplot(data[['UpperB', 'LowerB']], linestyle='dashdot'),

  mpf.make_addplot(data['PercentB'], panel='lower', color='g', secondary_y='auto'),

  mpf.make_addplot(data['Adj Close'], color='b', linestyle='dotted', secondary_y=True)]

  mpf.plot(data, type='candle', addplot=add_plot, volume=True, figscale=1.5, style='default')

  plt.show() # 显示

  得出结果

  这里需要注意的是style只能在plot方法中使用,不能在make_addplot方法中使用,而linestyle两者都可以用。

  二、marketcolors和mpf_style以及其他常用设置

  marketcolors设置图表颜色

  """

  make_marketcolors() 设置k线颜色

  :up 设置阳线柱填充颜色

  :down 设置阴线柱填充颜色

  :edge 设置蜡烛线边缘颜色 'i' 代表继承k线的颜色

  :wick 设置蜡烛上下影线的颜色

  :volume 设置成交量颜色

  :inherit 是否继承, 如果设置了继承inherit=True,那么edge即便设了颜色也会无效

  """

  my_color = mplfinance.make_marketcolors(up='cyan', down='red', edge='black', wick='black', volume='blue')

  添加图表样式

  这里的设置应该在plot方法之前设置,而设置完成marketcolors后需要使用mpf_style添加设置,最后plot()方法的关键字参数style=my_style 才会生效。下面添加mpf_style,最后一起展示效果。

  """

  make_mpf_style() 设置mpf样式

  :gridaxis:设置网格线位置,both双向

  :gridstyle:设置网格线线型

  :y_on_right:设置y轴位置是否在右

  """

  my_style = mplfinance.make_mpf_style(marketcolors=my_color, gridaxis='both', gridstyle='-.', y_on_right=True)

  这里要注意的是前面make_marketcolors的设置,必须这里作为关键字参数添加进来。其他几项常用设置:title:设置标题;ylabel 设置主图Y轴标题;ylabel_lower 设置成交量Y轴标题;savefig:导出图片路径以及文件名(包括后缀)。

  最后一步绘图:

  b_list, s_list = self.data_analyze(data)

  add_plot = [

  mplfinance.make_addplot(b_list, scatter=True, markersize=200, marker='*', color='y'),

  mplfinance.make_addplot(s_list, scatter=True, markersize=200, marker='s', color='r'),

  mplfinance.make_addplot(data[['UpperB', 'LowerB']], linestyle='dashdot'),

  mplfinance.make_addplot(data['PercentB'], panel='lower', color='g', secondary_y='auto'),

  mplfinance.make_addplot(data['Adj Close'], color='b', linestyle='dotted', secondary_y=True)]

  """

  plot绘图的部分参数

  :type设置图像类型'ohlc'/'candle'/'line/renko'

  :mav 绘制平局线

  :show_nontrading= True 显示非交易日(k线之间有间隔),False 不显示交易日,k线之间没有间隔

  :title:设置标题

  :ylabel=设置主图Y轴标题

  :ylabel_lower 设置成交量一栏Y坐标标题

  :figratio:设置图形纵横比

  :figscale 设置图像的缩小或放大,1.5就是放大50%,最大不会超过电脑屏幕大小

  :style 设置整个图表样式,可以使用前面设置的样式my_style,只能在plot函数中使用指定整个图表样式,不能在make_addplot中使用。

  savefig:导出图片,填写文件名及后缀

  """

  mplfinance.plot(data, type='candle', addplot=add_plot, volume=True, figscale=1.5, style=my_style,title='****报价', figratio=(5, 5), ylabel='价格', ylabel_lower='成交量', savefig='my_image.png')

  plt.show() # 显示

  最后得出结果。

原文地址:https://www.cnblogs.com/djw12333/p/12795084.html