为烛台系列添加标签

时间:2014-04-08 07:13:29

标签: python matplotlib candlestick-chart

以下代码段正在绘制4个烛台的两个烛台系列:

from pylab import * 
from matplotlib.finance import candlestick
import matplotlib.gridspec as gridspec

quotes = [(734542.0, 1.326, 1.3287, 1.3322, 1.3215), (734543.0, 1.3286, 1.3198, 1.3292, 1.3155), (734546.0, 1.321, 1.3187, 1.3284, 1.3186), (734547.0, 1.3186, 1.3133, 1.3217, 1.308)]
quotes2 = [(734542.0, 1.0, 0.9979, 1.0046, 0.9953), (734543.0, 0.998, 0.9991, 1.0024, 0.9952), (734546.0, 0.9991, 1.0014, 1.0038, 0.9951), (734547.0, 1.003, 1.0028, 1.0047, 1.0002)]

fig, ax = subplots()
candlestick(ax,quotes,width = 0.5, colorup = "green", colordown = "red")
candlestick(ax,quotes2, width = 0.2, colorup = "grey", colordown = "black")
ax.xaxis_date()
ax.autoscale_view()
ax.legend(loc=3)


plt.show()

我无法在两个系列中添加标签,而在网络上我还没有找到任何内容。什么是正确的语法来做到这一点?

candlestick(ax,quotes,width = 0.5, label = "Series 1") #but it doesn't work

注意:我需要的不是在特定点上注释某些内容,例如this或其他线程解释,而是从最终目的添加到图表图例中的正确标签是绘制几个标准化的价格序列,以便在视觉上相互比较。

ADD :更准确地说,"但它没有工作"粗略的尝试确实不会对以下预期错误起作用:

TypeError: candlestick() got an unexpected keyword argument 'label'

1 个答案:

答案 0 :(得分:1)

您需要更改一些地方才能获得标签:

C1=candlestick(ax,quotes,width = 0.5, colorup = "green", colordown = "red")
C2=candlestick(ax,quotes2, width = 0.2, colorup = "grey", colordown = "black")
ax.xaxis_date()
ax.autoscale_view()
ax.legend((C1[1][0],C2[1][0]), ('label1', 'label2'),loc=3)

enter image description here

问题是我们在这里有colorupcolordown,您无法轻易将它们放在图例中(嗯,您可能会,这将非常复杂)

那么为什么C2[1][0]?因为这是C1

In [5]:

C1
Out[5]:
([<matplotlib.lines.Line2D at 0x76b3c90>,
  <matplotlib.lines.Line2D at 0x759d3b0>,
  <matplotlib.lines.Line2D at 0x759dab0>,
  <matplotlib.lines.Line2D at 0x75a61d0>],
 [<matplotlib.patches.Rectangle at 0x76b3df0>,
  <matplotlib.patches.Rectangle at 0x759d590>,
  <matplotlib.patches.Rectangle at 0x759dc90>,
  <matplotlib.patches.Rectangle at 0x75a63b0>])

如果您有其他情节将遵循烛台图:

plt.hlines(1.10, plt.xlim()[0], plt.xlim()[1], label='Other Plot') #such as an horizontal line
#And may be other plots.
handles, labels = ax.get_legend_handles_labels()
import operator
hl = sorted(zip(handles, labels),
            key=operator.itemgetter(1)) #sort is optional
handles2, labels2 = zip(*hl)
handles2=list(handles2)+[C1[1][0],C2[1][0]] #put the candel plot legend to the end
labels2=list(labels2)+['label1', 'label2'] #put the candel plot legend to the end
ax.legend(handles2, labels2, loc=8)

enter image description here

取自documents