为蜡烛图系列添加标签
下面这段代码是在绘制两个蜡烛图系列,每个系列有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
注意:我需要的不是在某个特定点上添加注释,比如这个或其他帖子所说的那样,而是要在图表的图例中添加一个合适的标签,因为我的最终目的是绘制多个标准化的价格系列,以便进行视觉上的比较。
补充:更具体地说,"但它不工作"的粗略尝试确实没有成功,出现了以下预期的错误:
TypeError: candlestick() got an unexpected keyword argument 'label'
1 个回答
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)
问题是我们这里有 colorup
和 colordown
,你不能轻易把它们都放在图例里(其实可以,但会比较复杂)。
那么为什么是 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)
这段内容来自于 文档。