Как сохранить несколько графиков в виде отдельных файлов png с именами в Python?

Question

У меня есть код, который выглядит примерно так:

a = datetime(2019, 1, 22) #Key dates area
b = datetime(2019, 1, 23)

for _, d in df.set_index('Date').groupby('Country'):
    fig, ax = plt.subplots()
    d['Counts'].plot()
    plt.axhline(y=d['Counts'].mean(), color='r', linestyle='--')
    plt.xticks(rotation=90)
    plt.title(f"Weekly Sim Count for {d['Country'].iat[0]}")
    plt.xlabel('Week')


    plt.axvspan(a, b, color='gray', alpha=0.2, lw=0)
    plt.legend()
    plt.show()

Я надеюсь сохранить каждый график, возвращенный из этого l oop, как отдельные файлы png, где имя изображения основано на страна, представленная на графике. Есть ли способ сделать это?

Спасибо :)

Answer 1

То, что вы ищете, это plt.savefig , который сохраняет текущую цифру. Вы можете добавить его в l oop перед показом или вместо отображения графика:

# ...
plt.legend()
plt.savefig(f"{d['Country'].iat[0]}_plot.png")
# ...

, и он должен сохранить в текущем каталоге графики, созданные в l oop.