我试图在一维上绘制一个波函数,但它有实部和虚部,所以我做了一个3D绘图动画。这是一个屏幕截图:
我想做的主要事情是沿着x轴(现在是垂直的)展开它,这样看起来就不会被挤压。此外,最好将其设置为一组3个RGB轴,这些轴在点(0,0,0)处相交。在文档中,我找不到任何直接的方法来实现这一点。我正在附加用于设置动画的代码部分:
import matplotlib.pyplot as plt
import matplotlib.animation as animation
import mpl_toolkits.mplot3d.axes3d as p3
fig = plt.figure()
ax = fig.gca(projection='3d')
line, = ax.plot(REAL[0,:],IMAG[0,:],x,"r",linewidth=0.5)
def animacio(i):
ax.collections.clear()
line.set_data(REAL[i,:],IMAG[i,:])
line.set_3d_properties(x, 'z')
return line,
ani = animation.FuncAnimation(fig,animacio,interval=50, frames=Nt,repeat=True)
nom = 'Evolució_'
ani.save(str(nom)+'['+str(V0)+','+str(L)+','+str(l)+','+str(xi)+','+str(sigmax)+','+str(T)+']'+'.mp4', writer="ffmpeg", dpi=300)
plt.show()
print('Animation saved as: '+str(nom)+'['+str(V0)+','+str(L)+','+str(l)+','+str(xi)+','+str(sigmax)+','+str(T)+']'+'.mp4')
只需指定起点和终点并指定颜色,即可将彩色线添加到绘图中。“向上”轴的限制可通过
ax.set_zlim
设置。我创建了一条与你的曲线大致相似的演示曲线左侧为不带限制的绘图,右侧为带限制的绘图:
要获得更长的视图,您可以使用以下内容:
相关问题 更多 >
编程相关推荐