如何在Matplotlib中为图例行添加标题？|极客笔记

如何在Matplotlib中为图例行添加标题？

在数据可视化中，Matplotlib是一种常用的数据可视化工具，它为我们提供了许多绘图函数和可视化工具包，使得我们能够在Python中轻松绘制图形。

但有时候，我们需要在图例行中添加一些标题或注释，以便更好地解释图表的内容。那么，我们该怎么做呢？

本文将介绍如何在Matplotlib中为图例行添加标题。

创建示例数据

在正式开始讲解之前，我们需要先创建一些示例数据来进行演示。我们将创建一个简单的散点图，其中X轴为0到100的随机数据，Y轴为这些数据的平方。

import numpy as np
import matplotlib.pyplot as plt

# 创建示例数据
np.random.seed(42)
x = np.random.randint(0, 100, 50)
y = x**2

# 创建散点图
plt.scatter(x, y)

# 显示图表
plt.show()

现在我们已经创建好了一张简单的散点图，接下来将演示如何在图例行中添加标题。

添加图例

要添加图例，我们需要调用 plt.legend() 函数，并将其与 plt.scatter() 中的 label 参数配合使用，然后将 loc 参数设置为希望将图例放置在哪里。

import numpy as np
import matplotlib.pyplot as plt

# 创建示例数据
np.random.seed(42)
x = np.random.randint(0, 100, 50)
y = x**2

# 创建散点图，并添加图例
plt.scatter(x, y, label='Data points')
plt.legend(loc='upper left')

# 显示图表
plt.show()

现在，我们已经成功地为散点图添加了一个图例。

添加图例标题

默认情况下，图例行是没有标题的。但是，我们可以使用 plt.legend() 函数的 title 参数来为图例行添加标题。

import numpy as np
import matplotlib.pyplot as plt

# 创建示例数据
np.random.seed(42)
x = np.random.randint(0, 100, 50)
y = x**2

# 创建散点图，并添加图例和图例标题
plt.scatter(x, y, label='Data points')
plt.legend(title='Legend Title', loc='upper left')

# 显示图表
plt.show()

现在，我们已经成功地为图例行添加了一个标题。

修改图例标题样式

默认情况下，图例标题的颜色和字体与图例中的其他文本相同。但是，我们可以使用 plt.legend() 函数的 title_fontsize 和 title_fontweight 参数来修改图例标题的字体大小和权重。

import numpy as np
import matplotlib.pyplot as plt

# 创建示例数据
np.random.seed(42)
x = np.random.randint(0, 100, 50)
y = x**2

# 创建散点图，并添加图例和图例标题，修改标题样式
plt.scatter(x, y, label='Data points')
plt.legend(title='Legend Title', loc='upper left', title_fontsize=14, title_fontweight='bold')

# 显示图表
plt.show()

现在，我们已经成功地为图例行添加了一个修改后的标题。