Python plt 显示图例

在数据可视化过程中，图例是非常重要的元素之一。matplotlib中的pyplot模块提供了一些方法来显示图例。下面是一些常用的方法。

方法一：使用label和legend关键字参数

pyplot.plot()、pyplot.scatter()、pyplot.bar()等绘图函数都有一个关键字参数label，它用于指定图例中显示的文本内容。legend()方法用于在图中显示出所有的label。

示例：

import matplotlib.pyplot as plt

x = [1,2,3,4,5]
y = [1,4,9,16,25]

plt.plot(x, y, label="y=x^2")
plt.legend()
plt.show()

输出：

方法二：使用自定义图例

有些时候我们可能需要自己定义图例，比如显示颜色、形状等信息。这时可以使用legend()方法的handles和labels参数手动定义图例内容。

示例：

import matplotlib.pyplot as plt

x = [1,2,3,4,5]
y1 = [1,4,9,16,25]
y2 = [0,3,6,9,12]

plt.plot(x, y1, "r--", label="y=x^2")
plt.plot(x, y2, "g^", label="y=3x")
plt.legend(handles=[plt.Line2D([0], [0], linestyle="dashed", color="r", label="y=x^2"), 
                     plt.Line2D([0], [0], marker="^", color="g", label="y=3x")])
plt.show()

输出：

方法三：使用图例的loc参数

legend()方法还有一个参数loc，用于指定图例的位置。常用的取值包括：

• 0：自动分配
• 1：右上角
• 2：左上角
• 3：左下角
• 4：右下角
• 5：右侧

示例：

import matplotlib.pyplot as plt

x = [1,2,3,4,5]
y1 = [1,4,9,16,25]
y2 = [0,3,6,9,12]

plt.plot(x, y1, "r--", label="y=x^2")
plt.plot(x, y2, "g^", label="y=3x")
plt.legend(loc=2)
plt.show()

输出：

方法四：使用rcParams设置全局默认设置

如果我们需要在全局设置图例，可以使用rcParams这个字典对象来修改默认设置。比如，我们可以修改legend.fontsize的值来设置图例字体大小。

示例：

import matplotlib.pyplot as plt
import matplotlib as mpl

mpl.rcParams["legend.fontsize"] = 16

x = [1,2,3,4,5]
y1 = [1,4,9,16,25]
y2 = [0,3,6,9,12]

plt.plot(x, y1, "r--", label="y=x^2")
plt.plot(x, y2, "g^", label="y=3x")
plt.legend()
plt.show()

输出：

以上就是常用的显示图例的方法。希望对你的数据可视化有帮助！