在matplotlib中，我们常常需要绘制多个图形，或将多个图形组合在一起。在这种情况下，我们可以使用add_subplot()方法来创建子图。本文将从多个方面详细介绍如何使用add_subplot()方法。
一、subplot和plot的区别
在使用add_subplot()方法之前，我们先来了解一下subplot和plot的区别。
subplot是将多个图形分为若干个子图进行展示，而plot则是在一个图形中绘制不同数据。
subplot通过指定行数、列数和不同的位置参数，将图形分为多个子图。在每个子图中，我们可以使用plot()方法绘制需要展示的数据。
二、使用add_subplot()方法创建子图
在matplotlib中，我们可以使用add_subplot()方法创建子图。方法的参数表示行数，列数和位置顺序。
fig, axs = plt.subplots(2, 2) axs[0, 0].plot(x, y) axs[0, 1].scatter(x, y) axs[1, 0].bar(x, y) axs[1, 1].pie(y)
这段代码可以创建一个2x2的子图，并在每个子图中绘制不同的图形。
三、控制子图之间的间隔和大小
使用add_subplot()方法创建的子图之间会存在间隔。我们可以通过fig.subplots_adjust()方法来控制子图之间的间隔和大小。
fig, axs = plt.subplots(2, 2) fig.subplots_adjust(wspace=0.5, hspace=0.3) axs[0, 0].plot(x, y) axs[0, 1].scatter(x, y) axs[1, 0].bar(x, y) axs[1, 1].pie(y)
这段代码将子图之间的水平间距设置为0.5，垂直间距设置为0.3。
四、控制子图大小和位置
如果我们需要创建不同大小和位置的子图，可以使用add_subplot()方法的第四个参数。
fig = plt.figure()
ax1 = fig.add_subplot(2,2,1)
ax1.set_title('Plot 1')
ax2 = fig.add_subplot(2,2,2)
ax2.set_title('Plot 2')
ax2.set_xlabel('x label')
ax3 = fig.add_subplot(2,2,3)
ax3.set_title('Plot 3')
ax3.set_ylabel('y label')
ax4 = fig.add_subplot(2,2,4)
ax4.set_title('Plot 4')
ax4.set_xlabel('x label')
ax4.set_ylabel('y label')
plt.tight_layout()
在这段代码中，我们使用add_subplot()方法的第四个参数来控制子图的位置。第一个参数表示总行数，第二个参数表示总列数，第三个参数表示子图在位置顺序。
在这个例子中，我们创建了一个2x2的图形，并设置每个子图的标题，x轴标签和y轴标签。
五、添加注释和标签
我们可以在子图中添加注释和标签。使用annotate()方法可以在子图中添加注释，而使用xlabel()和ylabel()方法可以添加x轴和y轴标签。
fig, axs = plt.subplots(2, 2)
axs[0, 0].plot(x, y)
axs[0, 0].annotate('Maximum', xy=(3, 8), xytext=(4, 9), arrowprops=dict(facecolor='black', shrink=0.05))
axs[0, 0].set_xlabel('x label')
axs[0, 0].set_ylabel('y label')
axs[0, 1].scatter(x, y)
axs[0, 1].annotate('Minimum', xy=(4, 2), xytext=(5, 1), arrowprops=dict(facecolor='black', shrink=0.05))
axs[0, 1].set_xlabel('x label')
axs[0, 1].set_ylabel('y label')
axs[1, 0].bar(x, y)
axs[1, 0].set_xlabel('x label')
axs[1, 0].set_ylabel('y label')
axs[1, 1].pie(y)
axs[1, 1].set_xlabel('x label')
axs[1, 1].set_ylabel('y label')
plt.tight_layout()
在这段代码中，我们使用annotate()方法在子图中添加注释，并使用xlabel()和ylabel()方法添加x轴和y轴标签。
六、总结
使用add_subplot()方法可以在matplotlib中创建多个子图。我们可以控制子图之间的间距，大小和位置，并在子图中添加注释和标签。这让我们能够更好地展示数据和知识点。
