Python Matplotlib subplot函数详解：创建子图

使用 Matplotlib 除可以生成包含多条折线的复式折线图之外，它还允许在一张数据图上包含多个子图。

调用 subplot() 函数可以创建一个子图，然后程序就可以在子图上进行绘制。subplot(nrows, ncols, index, **kwargs) 函数的 nrows 参数指定将数据图区域分成多少行；ncols 参数指定将数据图区域分成多少列；index 参数指定获取第几个区域。

subplot() 函数也支持直接传入一个三位数的参数，其中第一位数将作为 nrows 参数；第二位数将作为 ncols 参数；第三位数将作为 index 参数。

下面程序示范了生成多个子图：

import matplotlib.pyplot as plt
import numpy as np

plt.figure()
# 定义从-pi到pi之间的数据，平均取64个数据点
x_data = np.linspace(-np.pi, np.pi, 64, endpoint=True)  # ①
# 将整个figure分成两行两列，第三个参数表示该图形放在第1个网格
plt.subplot(2, 2, 1)
# 绘制正弦曲线
plt.plot(x_data, np.sin(x_data))
plt.gca().spines['right'].set_color('none')
plt.gca().spines['top'].set_color('none')
plt.gca().spines['bottom'].set_position(('data', 0))
plt.gca().spines['left'].set_position(('data', 0))
plt.title('正弦曲线')

# 将整个figure分成两行两列，并将该图形放在第2个网格
plt.subplot(222)
# 绘制余弦曲线
plt.plot(x_data, np.cos(x_data))
plt.gca().spines['right'].set_color('none')
plt.gca().spines['top'].set_color('none')
plt.gca().spines['bottom'].set_position(('data', 0))
plt.gca().spines['left'].set_position(('data', 0))
plt.title('余弦曲线')

# 将整个figure分成两行两列，并该图形放在第3个网格
plt.subplot(223)
# 绘制正切曲线
plt.plot(x_data, np.tan(x_data))
plt.gca().spines['right'].set_color('none')
plt.gca().spines['top'].set_color('none')
plt.gca().spines['bottom'].set_position(('data', 0))
plt.gca().spines['left'].set_position(('data', 0))
plt.title('正切曲线')

plt.show()

上面程序多次调用 subplot() 函数来生成子图，每次调用 subplot() 函数之后的代码表示在该子图区域绘图。上面程序将整个数据图区域分成 2×2 的网格，程序分别在第 1 个网格中绘制正弦曲线，在第 2 个网格中绘制余弦曲线，在第 3 个网格中绘制正切曲线。

可能有读者感到疑问，plot() 函数不是用于绘制折线图的吗？怎么此处还可用于绘制正弦曲线、余弦曲线呢？

其实此处绘制的依然是折线图。看程序中的 ① 号代码调用 numpy 的 linspace() 函数生成了一个包含多个数值的列表，该数值列表的范围是从 -pi 到 pi，平均分成 64 个数据点，程序中用到的 numpy.sin()、numpy.cos()、numpy.tan() 等函数也返回一个列表：传入这些函数的列表包含多少个值，这些函数返问的列表也包含多少个值。

这意味着上面程序所给制的折线图会包含 64 个转折点，由于这些转折点非常密集，看上去显得比较光滑，因此就变成了曲线。

如果读者将程序中x_data=np.linespace(-np.pi,np.pi,64,endpoint = True)代码的 64 改为4、6等较小的数，将会看到程序绘制的依然是折线图。

运行上面程序，可以看到如图 1 所示的效果。
如图 1 所示的显示效果比较差，程序明明只要显示 3 个子图，但第 4 个位置被空出来了，能不能让某个子图占多个网格昵？答案是肯定的，程序做好控制即可。例如，将上面程序改为如下形式：

import matplotlib.pyplot as plt
import numpy as np

plt.figure()
# 定义从-pi到pi之间的数据，平均取64个数据点
x_data = np.linspace(-np.pi, np.pi, 64, endpoint=True)  # ①
# 将整个figure分成两行一列，第三个参数表示该图形放在第1个网格
plt.subplot(2, 1, 1)
# 省略绘制正弦曲线
...

# 将整个figure分成两行两列，并将该图形放在第4个网格
plt.subplot(223)
# 省略绘制余弦曲线
...

# 将整个figure分成两行两列，并该图形放在第4个网格
plt.subplot(224)
#省略绘制正切曲线
...

plt.show()

上面程序中第 8 行代码将整个区域分成两行一列，并指定子图占用第 1 个网格，也就是整个区域的第一行：第 13 行代码将整个区域分成两行两列，并指定子图占用第 3 个网格（注意不是第 2 个网络，因为第一个子图已经占用了第一行，对于两行两列的网格来说，第一个子图已经占用了两个网格，因此此处指定子图占用第 3 个网格，这意味着该子图在第二行第一格）；第 18 行代码将整个区域分成两行两列，并指定子图占用第 4 个网格，这意味着该子图会在第二行第二格。

运行上面程序，可以看到如图 2 所示的效果。
如果读者不想这么费劲来计算行、列，将上面程序改为如下形式：

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.gridspec as gridspec

plt.figure()
# 定义从-pi到pi之间的数据，平均取64个数据点
x_data = np.linspace(-np.pi, np.pi, 64, endpoint=True)  # ①

# 将绘图区域分成2行3列
gs = gridspec.GridSpec(2, 3)
# 指定ax1占用第一行（0）整行
ax1 = plt.subplot(gs[0, :])
# 指定ax1占用第二行（1）的第一格（第二个参数0代表）
ax2 = plt.subplot(gs[1, 0])
# 指定ax1占用第二行（1）的第二、三格（第二个参数0代表）
ax3 = plt.subplot(gs[1, 1:3])

# 绘制正弦曲线
ax1.plot(x_data, np.sin(x_data))
ax1.spines['right'].set_color('none')
ax1.spines['top'].set_color('none')
ax1.spines['top'].set_color('none')
ax1.spines['bottom'].set_position(('data', 0))
ax1.spines['left'].set_position(('data', 0))
ax1.set_title('正弦曲线')

# 绘制余弦曲线
ax2.plot(x_data, np.cos(x_data))
ax2.spines['right'].set_color('none')
ax2.spines['top'].set_color('none')
ax2.spines['bottom'].set_position(('data', 0))
ax2.spines['left'].set_position(('data', 0))
ax2.set_title('余弦曲线')

# 绘制正切曲线
ax3.plot(x_data, np.tan(x_data))
ax3.spines['right'].set_color('none')
ax3.spines['top'].set_color('none')
ax3.spines['bottom'].set_position(('data', 0))
ax3.spines['left'].set_position(('data', 0))
ax3.set_title('正切曲线')

plt.show()

上面程序中的第 10 行代码将绘图区域分成两行三列；第 12 行代码调用 subplot(gs[0,:])，指定 ax1子图区域占用第一行整行，其中第一个参数 0 代表行号，没有指定列范围，因此该子图在整个第一行；第 14 行代码调用 subplot(gs[1, 0])，指定 ax2 子图区域占用第二行的第一格，其中第一个参数 1 代表第二行，第二个参数 0 代表第一格，因此该子图在第二行的第一格；第 16 行代码调用 subplot(gs [1 , 1 : 3])，指定 ax3 子图区域占用第二行的第二格到第三格，其中第一个参数 1 代表第二行，第二个参数 1:3 代表第二格到第三格，因此该子图在第二行的第二格到第三格。

定义完 ax1、ax2、ax3 这 3 个子图所占用的区域之后，接下来程序就可以通过 ax1、ax2、ax3 的方法在各自的子图区域绘图了。运行上面程序，可以看到如图 3 所示的效果。