python的三维绘图

三维表面图

$\text{画出三维表面图}Z=\sin \left(\sqrt{x^2+y^2}\right)$

import pylab as plt
import numpy as np
ax=plt.axes(projection="3d")
X=np.arange(-6,6,0.25)
Y=np.arange(-6,6,0.25)
X,Y=np.meshgrid(X,Y)
Z=np.sin(np.sqrt(X**2+Y**2))
surf=ax.plot_surface(X,Y,Z,cmap="coolwarm")
plt.colorbar(surf)
plt.show()

三维曲线图

$\text{画出三维曲面图}Z=50\sin \left(x+y\right)$

import pylab as plt
import numpy as np
x=np.linspace(-4,4,100)
x,y=np.meshgrid(x,x)
z=50*np.sin(x+y)
ax=plt.axes(projection="3d")
ax.plot_surface(x,y,z,cmap="coolwarm")
plt.show()

三维曲线

$\text{画出三维曲线}x=s^2\sin s,y=s^2\cos s,s=s,s\in \left[-50,50\right]$

import pylab as plt
import numpy as np
s=np.linspace(-50,50,1000)
x=s**2*np.sin(s);y=s**2*np.cos(s)
ax=plt.axes(projection="3d")
ax.plot(x,y,s,"k")
plt.show()