matplotlib 三维可视化

matplotlib可以实现画曲面/等高线/投影等.

曲面可视化

# 导入相关的库
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

n = 64
x1 = np.linspace(-3,3,n)
y1 = np.linspace(-3,3,n)
x, y = np.meshgrid(x1, y1)
z= x ** 2 + y ** 2

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 三维曲面
surf = ax.plot_surface(x, y ,z, rstride=1, cstride=1, cmap=plt.get_cmap('rainbow'))

#等高线，其实就是投影,zdir代表了视角,offset表示离视角轴0点的距离
#z方向的等高线
contour = ax.contourf(x,y,z,zdir='z', offset=0, cmap=plt.get_cmap('rainbow'))

#设置坐标轴
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

# 添加colorbar
fig.colorbar(contour, shrink=0.5, aspect=10)

plt.show()

等高线

#导入模块
import numpy as np
import matplotlib.pyplot as plt

#建立步长为0.01，即每隔0.01取一个点
step = 0.01
x = np.arange(-10,10,step)
y = np.arange(-10,10,step)

#将原始数据变成网格数据形式
X,Y = np.meshgrid(x,y)
#写入函数，z是大写
Z = X** 2 + Y**2
#设置打开画布大小,长10，宽6

plt.figure(figsize=(10,6))
#填充颜色，f即filled
cset = plt.contourf(X,Y,Z, cmap='rainbow')

#画等高线
contour = plt.contour(X,Y,Z)
# 显示等高线数值
plt.clabel(contour,fontsize=10,colors='k')
plt.colorbar(cset, shrink=0.6, aspect=10)
plt.show()