Matplotlib：Python三维绘图

1、创建三维坐标轴对象Axes3D

创建Axes3D主要有两种方式，一种是利用关键字projection='3d'来实现，另一种是通过从mpl_toolkits.mplot3d导入对象Axes3D来实现，目的都是生成具有三维格式的对象Axes3D

#figure：新的画布
fig=plt.figure()
#axes：坐标轴
ax1=plt.axes(projection='3d')
#ax=fig.add_subplot(111,projection='3d')#画子图
plt.show()

#方法二，利用三维轴
fig=plt.figure()
ax2=Axes3D(fig)

 

 2、三维曲线和散点

#三维曲线和散点
z=np.linspace(0,13,1000)
x=5*np.sin(z)
y=5*np.cos(z)
zd=13*np.random.random(100)
xd=5*np.sin(zd)
yd=5*np.cos(zd)
ax1.scatter3D(xd,yd,zd,cmap='Blues')#绘制散点图
ax1.plot3D(x,y,z,'gray') #绘制空间曲线
plt.show()

 

 3、三维曲面

import matplotlib.pyplot as plt
import numpy as np

fig=plt.figure()
ax3=plt.axes(projection='3d')

#定义三维数据
XX=np.arange(-5,5,0.5)
YY=np.arange(-5,5,0.5)
X,Y=np.meshgrid(XX,YY)
Z=np.sin(X)+np.cos(Y)

#作图
ax3.plot_surface(X,Y,Z,cmap='rainbow')
ax3.contour(X,Y,Z,zdim='z',offset=-2,cmap='rainbow') #等高线图，要设置offset，为Z的最小值
plt.show()

 

 可以通过在plot_surface中设置参数rstride、cstride设置步长，来得到更加清晰细腻的图像。这两个参数分别表示在横竖方向上的采样步长，越小绘图越精细。

ax3.plot_surface(X,Y,Z,rstride=1,cstride=1,cmap='rainbow')