Python Subplot怎么用？

在Python中，subplot是matplotlib库中的一个函数，用于在一个图中创建多个子图。它可以在同一张画布上创建多个图像，并进行排列和布局。本文将详细介绍如何使用subplot函数创建子图和进行相关设置。

一、创建子图

1、首先，我们需要导入matplotlib.pyplot库，并创建一个画布和子图对象：

import matplotlib.pyplot as plt

fig = plt.figure()  # 创建画布
ax1 = fig.add_subplot(2, 2, 1)  # 创建子图，参数分别指定行数、列数和子图位置
ax2 = fig.add_subplot(2, 2, 2)
ax3 = fig.add_subplot(2, 2, 3)
ax4 = fig.add_subplot(2, 2, 4)

2、接下来，我们可以在每个子图上绘制图形。例如，我们可以在第一个子图上绘制折线图：

import numpy as np

x = np.linspace(0, 2*np.pi, 100)
y = np.sin(x)

ax1.plot(x, y)
ax1.set_title('Sin Function')  # 设置子图标题

3、重复上述步骤，在其他子图上绘制不同的图形：

ax2.scatter(x, y)
ax2.set_title('Scatter Plot')

ax3.bar(x, y)
ax3.set_title('Bar Plot')

ax4.hist(y, bins=20)
ax4.set_title('Histogram')

plt.show()  # 显示图像

二、子图布局

1、使用subplot函数的第一个参数指定子图的行数和列数，第二个参数指定当前子图的位置。例如，subplot(2, 2, 1)表示创建2行2列的子图，并将当前子图设置为第1个。

2、可以使用plt.subplot2grid函数进行更复杂的子图布局。它需要三个参数：(grid_size, loc, rowspan/colspan)。

ax1 = plt.subplot2grid((3, 3), (0, 0))  # 在一个3行3列的网格中，从(0, 0)位置开始绘制
ax2 = plt.subplot2grid((3, 3), (1, 0), colspan=2)  # 在第2行从(0, 0)位置开始绘制，占2列
ax3 = plt.subplot2grid((3, 3), (1, 2), rowspan=2)  # 在第2行从(2, 0)位置开始绘制，占2行
ax4 = plt.subplot2grid((3, 3), (2, 0))
ax5 = plt.subplot2grid((3, 3), (2, 1))

三、子图间距和比例

1、可以使用fig.subplots_adjust函数来调整子图的间距。它接受left、right、bottom、top参数，用于控制子图与画布边界的距离。

fig.subplots_adjust(left=0.1, right=0.9, bottom=0.1, top=0.9)

2、可以使用plt.subplots函数创建多个子图，并指定子图的比例。它可以接受参数sharex或sharey，用于控制子图之间的比例关系。例如，sharex=’col’表示每一列的子图共享x轴比例。

fig, axs = plt.subplots(2, 2, sharex='col', sharey='row')

axs[0, 0].plot(x, y)
axs[0, 0].set_title('Subplot 1')

axs[0, 1].scatter(x, y)
axs[0, 1].set_title('Subplot 2')

# 其他子图绘制代码

plt.show()

通过对subplot函数的使用和相关设置，我们可以方便地在一个图中创建多个子图，并进行布局和设置。这样可以更好地展示多个图像之间的关系和比较，提高数据可视化的效果和表达能力。