十九、子图

在这个 Matplotlib 教程中,我们将讨论子图。 有两种处理子图的主要方法,用于在同一图上创建多个图表。 现在,我们将从一个干净的代码开始。 如果你一直关注这个教程,那么请确保保留旧的代码,或者你可以随时重新查看上一个教程的代码。

首先,让我们使用样式,创建我们的图表,然后创建一个随机创建示例绘图的函数:

  1. import random
  2. import matplotlib.pyplot as plt
  3. from matplotlib import style
  4. style.use('fivethirtyeight')
  5. fig = plt.figure()
  6. def create_plots():
  7. xs = []
  8. ys = []
  9. for i in range(10):
  10. x = i
  11. y = random.randrange(10)
  12. xs.append(x)
  13. ys.append(y)
  14. return xs, ys

现在,我们开始使用add_subplot方法创建子图:

  1. ax1 = fig.add_subplot(221)
  2. ax2 = fig.add_subplot(222)
  3. ax3 = fig.add_subplot(212)

它的工作原理是使用 3 个数字,即:行数(numRows)、列数(numCols)和绘图编号(plotNum)。

所以,221 表示两行两列的第一个位置。222 是两行两列的第二个位置。最后,212 是两行一列的第二个位置。

  1. 2x2
  2. +-----+-----+
  3. | 1 | 2 |
  4. +-----+-----+
  5. | 3 | 4 |
  6. +-----+-----+
  7. 2x1
  8. +-----------+
  9. | 1 |
  10. +-----------+
  11. | 2 |
  12. +-----------+

译者注:原文此处表述有误,译文已更改。

译者注:221是缩写形式,仅在行数乘列数小于 10 时有效,否则要写成2,2,1

此代码结果为:

十九、子图 - 图1

这就是add_subplot。 尝试一些你认为可能很有趣的配置,然后尝试使用add_subplot创建它们,直到你感到满意。

接下来,让我们介绍另一种方法,它是subplot2grid

删除或注释掉其他轴域定义,然后添加:

  1. ax1 = plt.subplot2grid((6,1), (0,0), rowspan=1, colspan=1)
  2. ax2 = plt.subplot2grid((6,1), (1,0), rowspan=4, colspan=1)
  3. ax3 = plt.subplot2grid((6,1), (5,0), rowspan=1, colspan=1)

所以,add_subplot不能让我们使一个绘图覆盖多个位置。 但是这个新的subplot2grid可以。 所以,subplot2grid的工作方式是首先传递一个元组,它是网格形状。 我们传递了(6,1),这意味着整个图表分为六行一列。 下一个元组是左上角的起始点。 对于ax1,这是0,0,因此它起始于顶部。 接下来,我们可以选择指定rowspancolspan。 这是轴域所占的行数和列数。

  1. 6x1
  2. colspan=1
  3. (0,0) +-----------+
  4. | ax1 | rowspan=1
  5. (1,0) +-----------+
  6. | |
  7. | ax2 | rowspan=4
  8. | |
  9. | |
  10. (5,0) +-----------+
  11. | ax3 | rowspan=1
  12. +-----------+

结果为:

十九、子图 - 图2

显然,我们在这里有一些重叠的问题,我们可以调整子图来处理它。

再次,尝试构思各种配置的子图,使用subplot2grid制作出来,直到你感到满意!

我们将继续使用subplot2grid,将它应用到我们已经逐步建立的代码中,我们将在下一个教程中继续。