如何将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()之前，您必须保存该图