假设我有

import matplotlib.pyplot as plt

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

我们假设_data是长度为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_plot以便将绘制的图形放在位置索引中吗？ 如果可以，怎么办？

说实话，我不太了解plt对象在Matplotlib中的工作方式。