迹忆客 专注技术分享

本文介绍了如何使用 Python 中的 matplotlib.pyplot.figure.add_subplot() 方法为 Matplotlib 图添加子图。

使用 matplotlib.pyplot.figure.add_subplot() 方法在图中添加子图

import matplotlib.pyplot as plt 

fig=plt.figure(figsize=(5,5))

x=[1,2,3,4,5]
y=[2,4,5,1,2]

subplot=fig.add_subplot(1,1,1)
subplot.plot(x, y)

fig.suptitle("Figure with only one Subplot")
plt.show()

输出：

它生成的图中只有一个子图。该图可以解释为一个包含所有子图和图中其他元素的高级元素。我们首先初始化大小为 (5,5) 的图对象，然后使用 add_subplot() 方法向图中添加一个子图。add_subplot() 方法的第 1 个参数代表图中子图的行数，第 2 个参数代表子图的列数，第 3 个参数代表子图在网格中的位置。

这里，fig.add_subplot(1,1,1) 将在图中 1 X 1 网格的第一个位置添加一个子图。另外，我们也可以使用 fig.add_subplot(111) 来实现同样的结果。add_subplot() 方法返回一个 axes 对象，它可以用来绘制和添加子图的细节。

import matplotlib.pyplot as plt 

fig=plt.figure(figsize=(8,6))

ax_1=fig.add_subplot(121)
ax_1.text(0.3, 0.5, 'subplot(121)')

ax_2=fig.add_subplot(222)
ax_2.text(0.3, 0.5, 'subplot(222)')

ax_3=fig.add_subplot(224)
ax_3.text(0.3, 0.5, 'subplot(224)')

fig.suptitle("Figure with multiple Subplots")
plt.show()

输出：

这里，我们有一个包含三个子图的图。第一个子图是网格中的第 1 个数字，有 1 行 2 列。第 2 个子图代表网格中的第 2 个数字，有两行两列。同样，第三行代表网格中第四个位置的子图，有两行两列。

使用 Matplotlib.pyplot.subplots() 创建带有子图的图

Matplotlib.pyplot.subplots() 也可以给图添加子图。我们将行数和列数作为参数传递给该方法，该方法返回一个图对象和轴对象，可以用来操作图形。

import matplotlib.pyplot as plt 

fig,ax=plt.subplots(2,1)

ax[0].text(0.3,0.5,"1st Subplot")
ax[0].set_xticks([])
ax[0].set_yticks([])

ax[1].text(0.3,0.5,"2nd Subplot")
ax[1].set_xticks([])
ax[1].set_yticks([])

fig.suptitle('Figure with 2 subplots',fontsize=16) 
plt.show()

输出：

它使用 subplots() 方法在两行一列的网格中生成一个有两个子图的图形。第 1 个子图或顶部的轴可以用 ax 列表中的第 1 个元素进行操作，第 2 个子图或底部的轴可以用 ax 列表中的第 2 个元素进行操作。

import matplotlib.pyplot as plt 

x=[1,2,3,4,5]
y1=[i**2 for i in x ]
y2=[4*i+1 for i in x]

fig,ax=plt.subplots(2,1)

ax[0].plot(x,y1)
ax[0].set_xlabel("X")
ax[0].set_ylabel("Y")

ax[1].plot(x,y2)
ax[1].set_xlabel("X")
ax[1].set_ylabel("Y")

plt.tight_layout()
plt.show()

输出：

它生成一个有两个子图的图形，在上面的子图中绘制 x^2 函数，在下面的子图中绘制 4x+1 函数。