Python matplotlib绘图设置图例案例

下面我为大家详细讲解“Python matplotlib绘图设置图例案例”的完整攻略。

1. 前言

在使用Python的matplotlib库进行数据可视化时，我们经常需要在图中添加图例。图例可帮助我们说明数据的含义和不同线条或颜色的区分，使我们的图表更加清晰易懂。本文将介绍如何在matplotlib中设置图例，以及两个示例来演示如何在不同的情况下添加图例。

2. matplotlib中设置图例的常用方法

在matplotlib中设置图例时，以下三个参数是常用的：

• label：在绘图中给线条或标记命名，标记的名字将会在图例中显示；
• legend：在图中添加图例；
• handles：用来控制哪些对象将在图例中显示。

在绘制完图形后，通过调用ax.legend()或plt.legend()可以将设置好的图例添加到图中。

下面是一个简单的示例：

import matplotlib.pyplot as plt
import numpy as np

# 创建示例数据
x = np.linspace(0, 2 * np.pi, 50)
y1 = np.sin(x)
y2 = np.cos(x)

# 画出两条线
plt.plot(x, y1, label="sine")
plt.plot(x, y2, label="cosine")

# 添加图例
plt.legend()

# 显示图形
plt.show()

上述代码将绘制两条曲线，同时在图的右上角添加一个图例，图例中显示曲线的名称。

3. 示例1：设置图例位置

当我们需要将图例放置在不同位置时，matplotlib提供了几个常用的位置参数。下面是一个示例代码，演示如何设置图例的位置：

import matplotlib.pyplot as plt
import numpy as np

# 创建示例数据
x = np.linspace(0, 2 * np.pi, 50)
y1 = np.sin(x)
y2 = np.cos(x)

# 画出两条线
plt.plot(x, y1, label="sine")
plt.plot(x, y2, label="cosine")

# 添加图例，并设置位置为下方中间
plt.legend(loc="lower center")

# 显示图形
plt.show()

上述代码将绘制两条曲线，并将图例放置在图形的下方中间位置。可以通过指定loc参数来改变图例的位置，如loc='upper left'表示将图例放置在图形的左上角。

示例2：自定义图例样式

有时，我们可能会需要自定义图例的外观样式。此时，我们可以使用handles参数来手动创建图例对象，并通过它们的一些属性来控制图例的样式。下面是一个示例代码，演示如何自定义图例的样式：

import matplotlib.pyplot as plt
import numpy as np

# 创建示例数据
x = np.linspace(0, 2 * np.pi, 50)
y1 = np.sin(x)
y2 = np.cos(x)

# 画出两条线
line1, = plt.plot(x, y1, label="sine")
line2, = plt.plot(x, y2, label="cosine")

# 手动创建图例
leg = plt.legend(handles=[line1, line2], labels=["Sine Curve", "Cosine Curve"])
leg.get_lines()[0].set_linestyle('--')
leg.get_lines()[1].set_linestyle(':')

# 显示图形
plt.show()

上述代码将绘制两条曲线，并手动创建图例对象。我们通过set_linestyle()方法来自定义线条的样式。

4. 总结

在matplotlib中，我们可以通过简单的调用ax.legend()或plt.legend()来添加图例。并且，通过传递不同的参数，我们可以自定义图例的样式和位置。希望这篇文章能够帮助你在数据可视化中更好地使用matplotlib。