小编San*_*tos的帖子

我想绘制一个如下图所示的图表：

我编写了以下代码，为每个时间步绘制了波形图。

import matplotlib.pyplot as plt
import numpy as np

def u_0(x):
    a = 1.0/np.cosh(2.0*(x+8.0))
    b = 1.0/np.cosh((x+1.0))
    return 8.0*a*a+2.0*b*b

#spatial grid
N = 100
x = np.linspace(-10,10,N)

#time
Nt = 100
tlist = np.linspace(0.0,2.0,Nt)

#velocity
c = 5.0
count = 0
for t in tlist:
  u = u_0(x-c*t)
  plt.figure()
  plt.plot(x,u)
  plt.savefig(str(count))
  count = count+1
  plt.close()

如何将这些图片连接在一起并得到如图所示的图表？有没有标准的方法来做到这一点？