Python绘图标签

Question

我得画出10种不同的资料。目前，我正在手动输入标题栏的标题栏中的质量。

plt.plot(np.log(dist_2[1:]/var2['r200'][:20]), np.log(sigma_num_2),'b-o', color = 'b', label = "MASS1 = 7.6x10^13")

label是否只使用手动输入的字符串，或者是否有一种方法来指定label = mass，以便将变量(在本例中为质量)的值作为输入？

Answer 1

根据文档(api.html)：

标签字符串或任何具有‘%s’转换的可打印的内容。

因此，在您的示例中，要获得label = mass，您必须在需要时使用带有附加格式选项的label = "%.1E" % mass。

很可能您必须重新考虑您的mass变量。要将示例中手动键入的内容附加到数值中，还需要一个与MASS1等效的字符串，除非将大量值放入数组并在此数组上创建迭代图。在这种情况下，可以根据数组索引动态创建MASSX标签：

indexVal = 0
for massVal in mass: 
    indexVal += 1

    ...code for getting dist_2, var2, sigma_num_2 variables...

    plt.plot(np.log(dist_2[1:]/var2['r200'][:20]), np.log(sigma_num_2),'b-o', color = 'b', label = "MASS%s = %.1E" % (indexVal, massVal))

Answer 2

标签必须是字符串，格式化数字%e

plt.plot(..., label = "MASS1 = %.1e" % mass[0])

Answer 3

我认为最matplotlib的方式是在生成地块后发布一个单独的legend()。

l_plot=[]
for i in range(10):
    x=arange(10)
    y=random.random(10)
    l_plot.append(plt.plot(x, y, '+-'))
plt.xlim(0,12)
plt.legend([item[0] for item in l_plot], map(str, range(10))) #change it to the plot labels say ['Mass = %f'%item for item in range(10)].
plt.savefig('temp.png')

​

​