python - 如何从 matplotlib.figure.Figure 列表中保存 matplotlib 图？

Question

创建matplotlib图形后是否可以保存它？例如:

import matplotlib.pyplot as plt
import numpy
normal = numpy.random.normal(0, 1, 100)
df = pandas.DataFrame(normal.reshape((10, 10)))

figs = []

for i in df.keys():
    fig = plt.figure()
    plt.plot(df.index, df[i])
    figs.append(fig)

print(figs)

生成列表中的数字:

[<matplotlib.figure.Figure object at 0x7f8f158f1b00>, <matplotlib.figure.Figure object at 0x7f8f100d6128>, <matplotlib.figure.Figure object at 0x7f8f10081da0>, <matplotlib.figure.Figure object at 0x7f8f0a0354a8>, <matplotlib.figure.Figure object at 0x7f8f09fbd470>, <matplotlib.figure.Figure object at 0x7f8f09f18b00>, <matplotlib.figure.Figure object at 0x7f8f09f12d68>, <matplotlib.figure.Figure object at 0x7f8f09e88860>, <matplotlib.figure.Figure object at 0x7f8f09ebbc18>, <matplotlib.figure.Figure object at 0x7f8f09d6e198>]

我不想在创建后直接显式保存绘图，但可以选择稍后执行此操作(可能是在另一个类中)。理想的情况是将 fig 传递给 plt.savefig，但似乎没有可用于此类使用的插槽。

澄清一下。我想做的是:

fname = r'/path/to/file.jpg'
plt.savefig(fname, fig=figs[2])

这可以通过其他方式实现吗？

Answer 1

只需使用fig.savefig()。以下是您的案例的代码片段:

for idx, fig in enumerate(figs):
    fname = '{}_tmp.jpg'.format(idx)
    fig.savefig(fname)