python - 迭代子图

Question

目前，我制作了一个子图（使用matplotlib）

import matplotlib.pyplot as plt
fig, (ax1, ax2, ax3) = plt.subplots(3, sharex=True)

我想在我的图中有一个可变数量（n）的子图，名为 ax1, ax2, ... axn 所以它应该类似于

mylist = []
for num in xrange(n):
     mylist.append('ax'+str(num))
fig, (mylist) = plt.subplots(n, sharex=True)

但是，这不起作用，因为它只是覆盖“mylist”而不是使用列表中的名称。

另一种解决方案是，如果我没有创建列表而只使用它为轴创建的名称，但是，我不知道如何访问这些名称。如果我尝试使用

mylist[0]

控制台说

<matplotlib.axes._subplots.AxesSubplot object at 0x09BFAEF0>

如何使用此数字绘制一条线，例如

 ax1.plot(data[:,0], data[:,1]

score 1 · Accepted Answer

您要做的是动态创建变量。您可以使用以下代码来做到这一点：

vars()[str_name_of_your_variable] = variable_assignment

然而，这是非常不鼓励的，因为它没有提供处理可变数量数据的有效方法。最好将分配存储在列表或字典中：

fig, axes = plt.subplots(n, sharex=True)

在axes您拥有所有轴时，您可以通过选择它们axes[n]。如果需要，可以将其放入字典中。

score 0 · Accepted Answer

创建多个轴时：

fig, axes = plt.subplots(3)

axes将是一个包含 3 个轴对象的 numpy 数组。

为了使用它们进行绘图，您可以索引数组以访问轴对象并像往常一样进行绘图 ( ax.plot(...))：

axes[0].plot(x, y)

如果你有很多轴，那么你也可以遍历这些轴并绘制：

for ax in axes.flatten():
    ax.plot(x, y)

2 回答 2