Как использовать Matplotlib сохранить несколько графиков в файл в цикле

Question

x = [1 2 5 6 3 .....]
n = len(x)//34
i = 0
while i < n*34:
    fig, axs = plt.subplots(2)
    axs[0].plot(x[i:34+i],y[i:34+i],x_l[i:34+i],y_l[i:34+i]) #knee
    axs[1].plot(x_a[i:34+i],y_a[i:34+i],color = 'red') #ank
    axs[1].plot(x_l_a[i:34+i],y_l_a[i:34+i],color = 'green')
    axs[0].axis('off')
    axs[1].axis('off')
    plt.savefig('test')
    i = i + 17

Приведенный выше код сохранит только последний график в моем файле, но я хотел бы сгенерировать и сохранить несколько графиков для различных значений i.

Answer 1

В настоящее время вы перезаписываете один и тот же файл "test" на каждой итерации; таким образом, когда ваша программа завершает работу, у вас остается только последняя фигура, сохраненная на диск. Если вы хотите сохранить несколько файлов, вам нужно использовать разные имена файлов на каждой итерации в вашем вызове для savefig.

Например, вы можете использовать переменную итерации i и изменить свой вызов на:

plt.savefig("test{}".format(i))