为什么很多例子在 Matplotlib/pyplot/python 中使用 `fig, ax = plt.subplots()`

Question

我正在通过学​​习示例来学习使用matplotlib，很多示例似乎在创建单个图之前都包含如下一行...

fig, ax = plt.subplots()

这里有一些例子......

• Modify tick label text
• http://matplotlib.org/examples/pylab_examples/boxplot_demo2.html

我看到这个函数被大量使用，即使这个例子只是试图创建一个图表。有没有其他优势？ subplots() 的官方演示在创建单个图表时也使用了f, ax = subplots，之后它只引用了 ax。这是他们使用的代码。

# Just a figure and one subplot
f, ax = plt.subplots()
ax.plot(x, y)
ax.set_title('Simple plot')

Answer 1

plt.subplots() 是一个函数，它返回一个包含图形和坐标区对象的元组。因此，当使用fig, ax = plt.subplots() 时，您将此元组解包到变量fig 和ax 中。如果您想更改图形级别的属性或稍后将图形保存为图像文件（例如使用fig.savefig('yourfilename.png')），使用fig 非常有用。您当然不必使用返回的图形对象，但很多人稍后会使用它，因此很常见。此外，所有轴对象（具有绘图方法的对象）无论如何都有一个父图形对象，因此：

fig, ax = plt.subplots()

比这更简洁：

fig = plt.figure()
ax = fig.add_subplot(111)

Answer 2

这里只是补充。

下面的问题是，如果我想要图中更多的子图怎么办？

正如文档中提到的，我们可以使用fig = plt.subplots(nrows=2, ncols=2)在一个图形对象中设置一组带有grid(2,2)的子图。

那么我们知道，fig, ax = plt.subplots() 返回一个元组，我们先试试fig, ax1, ax2, ax3, ax4 = plt.subplots(nrows=2, ncols=2)。

ValueError: not enough values to unpack (expected 4, got 2)

它会引发错误，但不用担心，因为我们现在看到 plt.subplots() 实际上返回了一个包含两个元素的元组。第一个必须是图形对象，另一个应该是一组子图对象。

让我们再试一次：

fig, [[ax1, ax2], [ax3, ax4]] = plt.subplots(nrows=2, ncols=2)

并检查类型：

type(fig) #<class 'matplotlib.figure.Figure'>
type(ax1) #<class 'matplotlib.axes._subplots.AxesSubplot'>

当然，如果你使用参数 as (nrows=1, ncols=4)，那么格式应该是：

fig, [ax1, ax2, ax3, ax4] = plt.subplots(nrows=1, ncols=4)

所以只要记住保持列表的构造与我们在图中设置的子图网格相同。

希望这对你有帮助。

Answer 3

作为对问题和上述答案的补充，plt.subplots() 和plt.subplot() 之间还有一个重要区别，请注意末尾缺少的's'。

可以使用plt.subplots() 一次制作所有子图，并将子图的图形和轴（轴的复数）作为元组返回。可以将图形理解为画布的画布。

# create a subplot with 2 rows and 1 columns
fig, ax = plt.subplots(2,1)

然而，如果你想单独添加子图，你可以使用plt.subplot()。它只返回一个子图的轴。

fig = plt.figure() # create the canvas for plotting
ax1 = plt.subplot(2,1,1) 
# (2,1,1) indicates total number of rows, columns, and figure number respectively
ax2 = plt.subplot(2,1,2)

但是，plt.subplots() 是首选，因为它可以让您更轻松地直接自定义整个图

# for example, sharing x-axis, y-axis for all subplots can be specified at once
fig, ax = plt.subplots(2,2, sharex=True, sharey=True)

而使用plt.subplot()，则必须为每个轴单独指定，这可能会变得很麻烦。

Answer 4

除了上面的答案，你可以使用type(plt.subplots())来检查对象的类型，它返回一个元组，另一方面，type(plt.subplot())返回你无法解压的matplotlib.axes._subplots.AxesSubplot。

Answer 5

使用plt.subplots() 很受欢迎，因为它为您提供了一个 Axes 对象并允许您使用 Axes 接口来定义绘图。

替代方法是使用全局状态接口、plt.plot 等功能：

import matplotlib.pyplot as plt

# global state version - modifies "current" figure
plt.plot(...)
plt.xlabel(...)

# axes version - modifies explicit axes
ax.plot(...)
ax.set_xlabel(...)

那么为什么我们更喜欢 Axes？

• 它是可重构的 - 您可以将一些代码放入一个采用 Axes 对象的函数中，并且不依赖于全局状态
• 更容易过渡到具有多个子图的情况
• 一个一致/熟悉的界面，而不是在两个界面之间切换
• 获取matplotlib所有功能深度的唯一途径

全局状态版本的创建方式易于交互使用，并且是 Matlab 用户熟悉的界面，但在较大的程序和脚本中，此处概述的要点倾向于使用 Axes 界面。

有一篇 matplotlib 博客文章更深入地探讨了这个主题：Pyplot vs Object Oriented Interface

处理两个世界相对容易。例如，我们可以始终请求当前坐标区：ax = plt.gca() ("get current axes")。