详解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库。