极客笔记matplotlib如何在subplot中设置标题
在matplotlib中，我们可以通过subplot来创建多个子图，并为每个子图设置不同的标题。在本文中，我们将详细介绍如何在subplot中设置标题。
设置单个子图的标题
首先，我们来看如何为单个子图设置标题。我们可以使用set_title方法为子图设置标题，示例如下：
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.plot([1, 2, 3, 4], [1, 4, 9, 16])
ax.set_title('Subplot Title')
plt.show()
Output:
运行上面的代码，我们会得到一个包含一个子图的图表，并且子图的标题为“Subplot Title”。
设置多个子图的标题
接下来，我们将展示如何为多个子图设置不同的标题。我们可以使用set_title方法为每个子图设置不同的标题，示例如下：
import matplotlib.pyplot as plt
fig, axs = plt.subplots(2)
fig.suptitle('Subplot Titles')
axs[0].plot([1, 2, 3, 4], [1, 4, 9, 16])
axs[0].set_title('Subplot 1')
axs[1].plot([1, 2, 3, 4], [1, 2, 3, 4])
axs[1].set_title('Subplot 2')
plt.show()
Output:
运行上面的代码，我们会得到一个包含两个子图的图表，每个子图都有自己的标题。
使用循环设置子图标题
如果我们有很多子图需要设置标题，可以使用循环来简化代码。示例如下：
import matplotlib.pyplot as plt
fig, axs = plt.subplots(2, 2)
fig.suptitle('Subplot Titles')
for i, ax in enumerate(axs.flat):
ax.plot([1, 2, 3, 4], [1, 2, 3, 4])
ax.set_title(f'Subplot {i+1}')
plt.show()
Output:
运行上面的代码，我们会得到一个包含四个子图的图表，每个子图都有自己的标题。
小结
通过本文的介绍，我们学习了如何在matplotlib中通过subplot设置子图的标题。我们可以为单个子图设置标题，也可以为多个子图设置不同的标题，甚至可以使用循环简化代码。