matplotlib中的多个绘图，函数返回p

Question

假设我有：

import matplotlib.pyplot as plt

def generate_plot(data):
    plt.plot(...)
    return plt

我们假设∗u数据是长度为4的可数。你知道吗

if "__name__" = "__main__":
   _data = ....
   plt.figure(1)
   for data_i, index in _data, range(4):
       plt.subplot(2,2,index+1)
       ??????? <---- what goes here?

   plt.savefig(...)

我是否可以调用generate\ u plot，以便将绘制的图形放置在位置索引中？ 如果是，怎么做？你知道吗

老实说，我不太明白plt对象在Matplotlib中是如何工作的。你知道吗

Answer 1

你不想return plt。您可以将plt看作是访问matplotlib中所有绘图内容的一种方式，而不是“绘图对象”。先生成数据，然后绘制。比如说我有

import numpy as np
import matplotlib.pyplot as plt

def generate_data():
    return np.random.randint(10, size=10)

plt.figure(1)
for i in range(4):
    plt.subplot(2, 2, i + 1)
    plt.plot(generate_data())

plt.show()