Matplotlib：在两个不同的图形中绘制相同的图形，而无需写入两次“绘图（x， y）”行

Question

我有一个简单的代码，它在两个不同的图（图1和图2）中绘制了完全相同的东西。然而，我必须写一行ax？。绘制（x，y）两次，一次用于ax1，一次用于ax2。我怎么可能只有一个plot表达式（对于我更复杂的代码来说，有多个redondant表达式可能是一个麻烦的来源）。类似于ax1、ax2。绘图（x，y）？

import numpy as np
import matplotlib.pyplot as plt

#Prepares the data
x = np.arange(5)
y = np.exp(x)

#plot fig1
fig1 = plt.figure()
ax1 = fig1.add_subplot(111)

#plot fig2
fig2 = plt.figure()
ax2 = fig2.add_subplot(111)

#adds the same fig2 plot on fig1
ax1.plot(x, y)
ax2.plot(x, y)

plt.show()

Answer 1

在不了解matplotlib的情况下，您可以将所有轴（？）添加到列表中：

to_plot = []
to_plot.append(ax1)
...
to_plot.append(ax2)
...

# apply the same action to each ax
for ax in to_plot: 
    ax.plot(x, y)

然后，您可以添加任意数量的内容，每个内容都会发生相同的事情。

Answer 2

可以将每个轴添加到列表中，如下所示：

import numpy as np
import matplotlib.pyplot as plt

axes_lst = []    
#Prepares the data
x = np.arange(5)
y = np.exp(x)


#plot fig1
fig1 = plt.figure()
ax1 = fig1.add_subplot(111)
axes_lst.append(ax1)

#plot fig2
fig2 = plt.figure()
ax2 = fig2.add_subplot(111)
axes_lst.append(ax2)

for ax in axes_lst:
    ax.plot(x, y)

plt.show()

或者，您可以使用此不受支持的功能来提取pyplot中的所有图形。摘自https://stackoverflow.com/a/3783303/1269969

figures=[manager.canvas.figure
         for manager in matplotlib._pylab_helpers.Gcf.get_all_fig_managers()]
for figure in figures:
    figure.gca().plot(x,y)