python - 如何将matplotlib的plt对象函数传递给python中的另一个方法？

Question

假设我有一个可以使用 plt.show 生成的图形。现在我希望将这个 plt 对象传递给这样的通用函数 -

generate_figs(plt):
    frmt = ['jpg','png']
    for i in frmt:
        plt.savefig('name.{}'.format(i))

generate_figs(plt)

这样的事能做吗？

Answer 1

这取决于你有多少数字。如果只有一个图形，您可以简单地(大部分)保持函数原样，因为 plt.savefig 将保存当前图形。

def generate_figs():
    frmt = ['jpg','png']
    for i in frmt:
        plt.savefig('name.{}'.format(i))

如果你有多个图形，那么你可以将特定的图形对象传递给函数并使用fig.savefig

import matplotlob.pyplot as plt

def generate_figs(fig):
    frmt = ['jpg','png']
    for i in frmt:
        fig.savefig('name.{}'.format(i))

fig1 = plt.figure()
plt.plot(some_data)

fig2 = plt.figure()
plt.plot(some_other_data)

generate_figs(fig1)

plt.show()

注意:您必须在调用 plt.show()

之前保存图形