十九、子图
在这个 Matplotlib 教程中,我们将讨论子图。 有两种处理子图的主要方法,用于在同一图上创建多个图表。 现在,我们将从一个干净的代码开始。 如果你一直关注这个教程,那么请确保保留旧的代码,或者你可以随时重新查看上一个教程的代码。
首先,让我们使用样式,创建我们的图表,然后创建一个随机创建示例绘图的函数:
import random
import matplotlib.pyplot as plt
from matplotlib import style
style.use('fivethirtyeight')
fig = plt.figure()
def create_plots():
xs = []
ys = []
for i in range(10):
x = i
y = random.randrange(10)
xs.append(x)
ys.append(y)
return xs, ys
现在,我们开始使用add_subplot
方法创建子图:
ax1 = fig.add_subplot(221)
ax2 = fig.add_subplot(222)
ax3 = fig.add_subplot(212)
它的工作原理是使用 3 个数字,即:行数(numRows
)、列数(numCols
)和绘图编号(plotNum
)。
所以,221 表示两行两列的第一个位置。222 是两行两列的第二个位置。最后,212 是两行一列的第二个位置。
2x2:
+-----+-----+
| 1 | 2 |
+-----+-----+
| 3 | 4 |
+-----+-----+
2x1:
+-----------+
| 1 |
+-----------+
| 2 |
+-----------+
译者注:原文此处表述有误,译文已更改。
译者注:
221
是缩写形式,仅在行数乘列数小于 10 时有效,否则要写成2,2,1
。
此代码结果为:
这就是add_subplot
。 尝试一些你认为可能很有趣的配置,然后尝试使用add_subplot
创建它们,直到你感到满意。
接下来,让我们介绍另一种方法,它是subplot2grid
。
删除或注释掉其他轴域定义,然后添加:
ax1 = plt.subplot2grid((6,1), (0,0), rowspan=1, colspan=1)
ax2 = plt.subplot2grid((6,1), (1,0), rowspan=4, colspan=1)
ax3 = plt.subplot2grid((6,1), (5,0), rowspan=1, colspan=1)
所以,add_subplot
不能让我们使一个绘图覆盖多个位置。 但是这个新的subplot2grid
可以。 所以,subplot2grid
的工作方式是首先传递一个元组,它是网格形状。 我们传递了(6,1)
,这意味着整个图表分为六行一列。 下一个元组是左上角的起始点。 对于ax1
,这是0,0
,因此它起始于顶部。 接下来,我们可以选择指定rowspan
和colspan
。 这是轴域所占的行数和列数。
6x1:
colspan=1
(0,0) +-----------+
| ax1 | rowspan=1
(1,0) +-----------+
| |
| ax2 | rowspan=4
| |
| |
(5,0) +-----------+
| ax3 | rowspan=1
+-----------+
结果为:
显然,我们在这里有一些重叠的问题,我们可以调整子图来处理它。
再次,尝试构思各种配置的子图,使用subplot2grid
制作出来,直到你感到满意!
我们将继续使用subplot2grid
,将它应用到我们已经逐步建立的代码中,我们将在下一个教程中继续。