详解plt.subplot_千锋教育

一、概述

plt.subplot是matplotlib库中的一个函数，它的作用是在单个图像中创建多个子图。通过自定义子图位置，我们可以将多组数据在同一个图像中展示出来，从而更直观地比较各组数据的差异。这一功能在数据分析与数据可视化中有着广泛的应用。

二、使用方法

在使用plt.subplot之前，需要先导入matplotlib库。plt.subplot函数共有三个参数：子图行数、子图列数和子图位置。其中，子图位置是一个整数，其取值范围为1至子图行数与子图列数的积。


import matplotlib.pyplot as plt
plt.figure(figsize=(8,6)) # 画布大小
plt.subplots_adjust(hspace=0.4) # 子图间距
plt.subplot(2,2,1) # 创建第1个子图
plt.plot([0,1], [0,1])
plt.title('subplot 1')
plt.subplot(2,2,2) # 创建第2个子图
plt.plot([1,0], [0,1])
plt.title('subplot 2')
plt.subplot(2,1,2) # 创建第3个子图
plt.plot([0,1], [0,1])
plt.title('subplot 3')
plt.show()

运行以上代码，即可创建一个包含3个子图的图像。其中第1和第2个子图都位于第1行，第3个子图位于第2行。

三、子图位置的设定

除了直接指定整数作为子图位置，我们还可以使用子图位置的另一种设定方式——使用plt.subplots中的gridspec参数。gridspec参数是一个2元素元组，分别指定子图行数和列数。通过调用gridspec子命令来设置子图位置。代码如下：


import matplotlib.gridspec as gridspec
import matplotlib.pyplot as plt
plt.figure(figsize=(8,6))
gs = gridspec.GridSpec(2,2) # 两行两列
ax1 = plt.subplot(gs[0,0]) # 第1个子图
ax1.plot([0,1], [0,1])
ax1.set_title('subplot 1')

ax2 = plt.subplot(gs[0,1]) # 第2个子图
ax2.plot([1,0], [0,1])
ax2.set_title('subplot 2')

ax3 = plt.subplot(gs[1,:]) # 第3个子图
ax3.plot([0,1], [0,1])
ax3.set_title('subplot 3')
plt.show()

运行以上代码，同样可以创建一个有3个子图的图像。不同之处在于，这次我们使用了一种更灵活的设定方法，即通过gridspec.GridSpec创建网格，通过plt.subplot指定子图位置。

四、多行多列图形

除了使用plt.subplot来创建子图，我们还可以使用plt.subplots来一次性创建多行多列的图形。代码如下：


import matplotlib.pyplot as plt
fig,axes = plt.subplots(nrows=2, ncols=2, figsize=[8,6])
axes[0,0].plot([0,1], [0,1])
axes[0,0].set_title('subplot 1')
axes[0,1].plot([1,0], [0,1])
axes[0,1].set_title('subplot 2')
axes[1,0].plot([0,1], [1,0])
axes[1,0].set_title('subplot 3')
axes[1,1].plot([0,1], [0,0])
axes[1,1].set_title('subplot 4')
plt.show()

这段代码创建了一个2×2的图形，每个子图都有一个对应的axes变量名用于进行进一步的设定。

五、子图刻度的设定

在使用plt.subplot或plt.subplots创建子图后，我们还希望设定每个子图的横、纵坐标轴上的刻度。这可以通过子图对象的axis()方法或set_xticks()、set_yticks()方法实现。代码如下：


import matplotlib.pyplot as plt
plt.figure(figsize=(8,6))
plt.subplots_adjust(hspace=0.4)

plt.subplot(2,2,1)
plt.plot([0,1], [0,1])
plt.title('subplot 1')
plt.xticks([0,0.5,1]) # 水平刻度线
plt.yticks([0,0.5,1]) # 垂直刻度线

plt.subplot(2,2,2)
plt.plot([1,0], [0,1])
plt.title('subplot 2')
plt.xticks([0,0.5,1]) # 水平刻度线
plt.yticks([0,0.5,1]) # 垂直刻度线

plt.subplot(2,1,2)
plt.plot([0,1], [0,1])
plt.title('subplot 3')
plt.xticks([0,0.5,1]) # 水平刻度线
plt.yticks([0,0.5,1]) # 垂直刻度线
plt.show()

运行以上代码，即可看到每个子图都有自己的刻度线。

六、子图标题、图形标题和轴标签

除了刻度线外，我们还可以通过子图对象的set_xlabel()、set_ylabel()和set_title()方法来设定自己的轴标签和标题。代码如下：


import matplotlib.pyplot as plt
plt.figure(figsize=(8,6))
plt.subplots_adjust(hspace=0.7)

plt.subplot(2,2,1)
plt.plot([0,1], [0,1])
plt.title('subplot 1')
plt.xlabel('x1')
plt.ylabel('y1')

plt.subplot(2,2,2)
plt.plot([1,0], [0,1])
plt.title('subplot 2')
plt.xlabel('x2')
plt.ylabel('y2')

plt.subplot(2,1,2)
plt.plot([0,1], [0,1])
plt.title('subplot 3')
plt.xlabel('x3')
plt.ylabel('y3')
plt.show()

运行以上代码，即可看到每个子图都有自己的轴标签和标题。

七、小结

本文详细解释了plt.subplot函数的使用方法和设定技巧，并通过代码示例进行了演示。希望本文能够对读者有所帮助，激发大家在数据分析和可视化中更加灵活地利用matplotlib库。