python - 将多条线绘制到单个 Python 图形上

Question

这是我的第一个 Stack Overflow 问题，如果该问题的格式不符合 Stack Overflow 社区的惯例，请原谅。

我的问题是如何(如果可能)使用 pyplot 在现有的 Python 图形中添加一条线？本质上，我想要完成的是我想要编译并运行我的代码，并生成一个带有线条的图形。然后，我想做的是更改代码中的一些数据，再次编译并运行代码，然后将运行代码生成的行显示在已经存在的图形上。

fig=plt.figure(figsize=(20,10))
time= np.array(range(0, time_steps+1)) 
sigma_11=Q[:, :, :, 0]
sigma_22=Q[:, :, :, 1]
sigma_12=Q[:, :, :, 2]
vel_x=Q[:, :, :, 3]
vel_y=Q[:, :, :, 4]
stiffness_mat=np.mat([[lamda+2*mu, lamda, 0],
                    [lamda, lamda+2*mu, 0],
                    [0, 0, 2*mu]])

Energ_P=np.zeros((time_steps+1))  
Energ_K=np.zeros((time_steps+1))
Energy_tot=np.zeros((time_steps+1))     
    for n in  range(0, time_steps+1):
Energ_P[n]=(Energy_P(stiffness_mat, Q[:, :, n, 0:3]))
Energ_K[n]=(Energy_K(Q[:, :, n, 3:5]))
Energy_tot[n]=Energ_P[n]+Energ_K[n]

sub1=fig.add_subplot(1, 2, 1)  
sub1.set_title('stress-time')
sub1.plot(time*del_t, sigma_11[bar_cells_x, bar_cells_y, :], label='stress_1')
sub1.plot(time*del_t, sigma_11[bar_cells_x/2, bar_cells_y/2, :], label='stress')
sub1.set_ylabel('sigma_11 [Pa]')
sub1.set_xlabel('time [s]')
sub1.legend(loc='best', fontsize='10')

plt.tight_layout()
plt.show()

我还没有真正尝试过任何方法来解决这个问题，除了多次运行代码之外。然而，当我这样做时，我最终所做的只是生成多个数字。

Answer 1

尝试将 num 参数添加到 plt.figure 中。换句话说，将 fig=plt.figure(figsize=(20,10)) 替换为 fig=plt.figure(num=1, Figsize=(20,10)).

来自文档，

num : integer or string, optional, default: none

If not provided, a new figure will be created, and the figure number will be incremented.