for循环中的绘图覆盖

Question

我正在使用FeniCS在不同的时间步求解PDE，然后将其存储到各种列表中，并使用matplotlib在python中绘制。我在尝试创建和保存循环中的多（三）个绘图时遇到问题。我只能保存一个绘图而不覆盖它们。忽略必要的细节，我的代码如下所示

for n in range(num_steps):

    #Update current time
    t += dt
    

    #Solve
    solve(a_form == L_form, u)
    
    #Store times
    t_vals.append(t)

    #Solve PDE, gives solution u
    solve(u)

    #Create empty lists
    u_vals_x = []
    u_vals_y = []
    u_vals_z = []

    #Set constant
    xyz_fixed_density = 1000
    #Store u values varying x, y and z held equal to 1
    for n in  np.linspace(x0,x1,xyz_fixed_density):
     u_vals_x.append(u(n,1,1))

     #Store u values varying y, x and z held equal to 1
    for n in  np.linspace(y0,y1,xyz_fixed_density):
     u_vals_y.append(u(1,n,1))


     #Store u values varying z, x and y held equal to 1
    for n in  np.linspace(z0,z1,xyz_fixed_density):
     u_vals_z.append(u(1,1,n))
    
    #First plot
    plt.scatter(np.linspace(x0,x1,xyz_fixed_density),u_vals_x,s=1)
    plt.legend(t_vals)
    plt.xlabel('$x$')
    plt.ylabel('$u(t,x,1,1)$')
    plt.savefig('u_vs_x.png')
    
    #Second plot
    plt.scatter(np.linspace(y0,y1,xyz_fixed_density),u_vals_y,s=1)
    plt.legend(t_vals)
    plt.xlabel('$y$')
    plt.ylabel('$u(t,1,y,1)$')
    plt.savefig('u_vs_y.png')
    
    #Third plot
    plt.scatter(np.linspace(z0,z1,xyz_fixed_density),u_vals_z,s=1)
    plt.legend(t_vals)
    plt.xlabel('$z$')
    plt.ylabel('$u(t,1,1,z)$')
    plt.savefig('u_vs_z.png')

这可能是一个简单的修复，但我似乎无法让它工作。提前谢谢

Answer 1

使用当前迭代（n）作为文件名的一部分；e、 g.更换

plt.savefig('u_vs_x.png')

与

plt.savefig(f'u_vs_x_{n}.png')

这将使用f字符串语法格式化代码。如果您使用的是不支持f字符串的较旧Python版本，请显式使用format：

plt.savefig('u_vs_x_{}.png'.format(n))

您还需要每次创建一个新绘图，例如通过

plt.figure()