在一个python脚本中将多个图保存到不同的pdf文件

Question

Here's the current code I have. It's supposed to plot multiple plots onto 2 different graphs, which it does correctly (the outputs of plt.show() are as they should be), but when I go to check the files generated, the same figure is saved in both graph1.pdf and graph2.pdf , and it's the one that should only be in graph1.pdf . I tried using pdf.clf() before saving the second plot and that only results in the said plot saving blank while the first still saves correctly. Any idea what I'm missing?

from matplotlib import pyplot as plt
from scipy.integrate import odeint as odin
import numpy as np

pdf = plt.figure()


phi_w_0 = [1.5,0]
damp = [0.02 , 0.05 , 0.1 , 0.2]

t = np.linspace(0,200,1000)

def f(w,t):
    return [w[1] , -(B*w[1]) + np.sin(w[0])] 

for B in damp:
    result = odin(f , phi_w_0 , t)
    plt.plot(t , result[:,1] , label='$ \\beta = $ %1.2lf ' % B)

plt.legend()
plt.xlabel('$ \\tau $')
plt.ylabel('$ \\phi $')
plt.show()
pdf.savefig("graph1.pdf")

plt.close()

def g(w,t): 
    return [w[1] , -(B*w[1]) + np.sin(w[0])]

for B in damp:
    result = odin(g , phi_w_0 , t)
    plt.plot(t , result[:,1] , label='$ \\beta = $ %1.2lf ' % B)

plt.legend()
plt.xlabel('$ \\tau $')
plt.ylabel('E')
plt.show()
pdf.savefig("graph2.pdf")

Answer 1

pdf.clf()
pdf = plt.figure()

solved it. Turns out I was on the right track with thinking I have to clear it, but I didn't realize I have to call it again afterwards. Thank you!