Активировать фигуру в матплотлиб

Question

Это кажется простым, но я не смог найти никакого решения для открытия нескольких фигур и сохранения их по имени. Я ищу что-то вроде этого:

fig1, ax1 = pl.subplots(1)
fig2, ax2 = pl.subplots(1)
...
pl.savefig('f1.png', fig1)
pl.savefig('f2.png', fig2)

обычно pl.savefig действует на последнюю активную фигуру. Итак, как можно активировать фигуру и сохранить ее, а затем повторить процесс для остальных фигур?

Answer 1

Вы можете сохранить изображение, используя сам объект рисунка:

fig1.savefig(...)

Кроме того, вы можете изменить текущую фигуру, вызвав plt.figure(1), чтобы выбрать первую созданную фигуру, а затем использовать plt.savefig(). Или вы можете использовать plt.figure(fig1.number) для переключения фокуса на fig1

import matplotlib.pyplot as plt

fig1, ax1 = plt.subplots(1)
fig2, ax2 = plt.subplots(1)

# Can choose one of the below to change the current figure
plt.figure(1)
# plt.figure(fig1.number)

plt.savefig(...) # will save fig1