Pyplot - 标签图

Question

我在尝试用 pyplot 标记数据时遇到了一场噩梦。

我目前正在绘制所有数据，如下所示：plt.plot(data)。

数据是一个数组，其中有一列是总成本，然后是其他列的子成本

我想添加一个图例并适当地标记每个数据列表。但我似乎无法让传奇发挥作用。 plt.legend(('Column 1','Column 2','etc.) , loc =1) 和其他类似的东西不起作用。如果我单独绘制每一列，它会由于某种原因停止正常工作。

我期待并感谢您的任何建议。

编辑：

for i in range(1,days):
    data.append(totalCost(i)) #cost returns retVal, construction, gas, and wage
plt.ylabel('Cost in US Dollars') 
plt.title('Economic Cost over Time')

plt.plot(data)
plt.legend(('Total','Construction','Gas','Wage'),loc=1) # Legend is blank

编辑2： 我没有将代码放在程序中的不同位置，而是重新组织了代码并将所有内容集中起来。随着这些变化，传奇开始发挥作用，一切似乎都得到了解决。我不知道问题是什么。

plt.plot(data)
plt.legend(('Total Cost', 'Construction Cost', 'Gas Cost','Wage Cost'),loc=0)
plt.grid(False)
plt.xlabel('Time (Days)')
plt.ylabel('Cost in US Dollars')
plt.title('Economic Cost over Time')
print ('   Close the Graph to Continue Using this Model')
plt.show()

Answer 1

您的代码在 Mac OS X 上使用 Matplotlib 1.1 可以正确显示图例，前提是：

• Matplotlib 在开头置于交互模式： plt.ion()
• 代码以 plt.show 结尾()，例如。

升级 Matplotlib 也可能有帮助。

最后一种可能的解决方案是确实使用自己的标签单独绘制每条线： plt.plot(, label='Construction') 等，并绘制图例与 plt.legend(loc='best')。

Answer 2

在 plt.legend() 中传递标题，您可以创建一个图例列表并将它们传递到您的绘图，即，
legends = ['总成本', '建筑成本', '天然气成本','工资成本']
plt.plot（数据，标签=图例）
plt.legend()