importnumpyasnpimportmatplotlib.pyplotaspltnzs=100z=np.linspace(0,1,nzs)kvec=np.array([0,2,10,30,100])cs=np.zeros((len(z),len(kvec)))foriinrange(len(kvec)):k=kvec[i]ifk==0:cs[:,i]=1-zelse:cs[:,i]=np.sinh(np.sqrt(k)*(1-z))/np.sinh(np.sqrt(k))plt.figure()plt.plot(z,cs)nk=3k=kvec[nk]tvec=np.array([0,0.001,0.005,0.01,0.05,0.1,1.0])nts=len(tvec)nterms=25c=np.zeros((nzs,nts))forjinrange(nts):t=tvec[j]sum_=np.zeros(nzs)forninrange(1,nterms+1):n2p2=n**2*np.pi**2+kterm=(-1)**n*(n/n2p2)*np.sin(n*np.pi*(1-z))*np.exp(-n2p2*t)sum_+=termc[:,j]=cs[:,nk]+sum_*2*np.piplt.figure()plt.plot(z,c)plt.show(block=False)# store datastore=np.column_stack((z,cs))withopen("csz.dat","w")asf:np.savetxt(f,store,fmt='%f',header='store')