来自函数的 matplotlib.pyplot 中的子图

Question

我尝试使用一些定义的函数来创建子图，该函数直接返回图。但我无法弄清楚为什么这不起作用。例如：

我的绘图功能类似于：

def plot_data(data):
    plt.plot(data)
    return plt.show()

假设我的数据是：

data1 = np.random.rand(50)
data2 = np.random.rand(50)
data3 = np.random.rand(50)
data4 = np.random.rand(50)

我正在尝试使用以下方法创建子图：

fig, ax = plt.subplots(nrows=2, ncols=2)

plt.subplot(2, 2, 1)
plot_data(data)

plt.subplot(2, 2, 2)
plot_data(data)

plt.subplot(2, 2, 3)
plot_data(data)

plt.subplot(2, 2, 4)
plot_data(data)

plt.tight_layout()
plt.show()

它返回：

Answer 1

只需删除 plot_data() 函数中的 return plt.show() 即可。这样就行了。

在这种情况下，即使使用那个额外的函数plot_data() 也可能没用，你可以直接使用plt.plot(data) :-)