把一组图放在一起,使用subplot函数来绘制子图。
plt.subplot(nrows, ncols, index, **kwargs)
- 第一个参数代表子图的行数;
- 第二个参数代表该行图像的列数;
- 第三个参数代表每行的第几个图像。
可以使用三个整数,或者一个三位整数来描述子图的位置信息。
子图将分布在行列的索引位置上。索引从1开始。
举个列子:plt.subplot(2, 3, 5) 和 plt.subplot(235) 是一样一样的。需要注意的是所有的数字不能超过10。
规则划分
import matplotlib
import matplotlib.pyplot as plt
np.random.seed(42)
t = np.arange(0, 20, 0.1)
nse1 = np.random.randn(len(t))
s1 = np.sin(5 * np.pi * 20 * t) + nse1
for i,color in enumerate("rgby"):
plt.subplot(221+i, facecolor=color)
plt.plot(t, s1) #横轴与纵轴数据
plt.show()
不规则划分
画一个3副图的图像,将整个表按照 2*2 划分 。
前两个分别是 (2, 2, 1) 和 (2, 2, 2)
第三个图占用了 (2, 2, 3) 和 (2, 2, 4)
显示需要对其重新划分, 按照 2 * 1 划分
前两个图占用了 (2, 1, 1) 的位置
因此第三个图占用了 (2, 1, 2) 的位置
import numpy as np
import matplotlib.pyplot as plt
def f(t):
return np.exp(-t) * np.cos(2*np.pi*t)
t1 = np.arange(0.0, 5.0, 0.1)
t2 = np.arange(0.0, 5.0, 0.02)
plt.figure(1)
plt.subplot(221)
plt.plot(t1, f(t1), 'bo', t2, f(t2), 'k')
plt.subplot(222)
plt.plot(t2, np.cos(2*np.pi*t2), 'r--')
plt.subplot(212)
plt.plot([1, 2, 3, 4], [1, 4, 9, 16])
plt.show()