Matplotlib系列4-生成3D三维曲面图

使用matplotlib可以生成漂亮精美，印刷级别的三维曲面图，只需要几行代码即可。下面我们直接看源代码

一 源代码

import numpy as np  
import matplotlib.pyplot as plt  
from mpl_toolkits.mplot3d import Axes3D  
  
# 创建一个新的figure  
fig = plt.figure()  
# 添加一个3d坐标轴  
ax = fig.add_subplot(111, projection='3d')  
  
# 定义数据  
X = np.linspace(-5, 5, 100)  
Y = np.linspace(-5, 5, 100)  
X, Y = np.meshgrid(X, Y)  
Z = np.sin(np.sqrt(X**2 + Y**2))  
  
# 绘制曲面图  
surf = ax.plot_surface(X, Y, Z, cmap='viridis')  
  
# 添加颜色条  
fig.colorbar(surf)  
  
# 设置坐标轴标签  
ax.set_xlabel('X Label')  
ax.set_ylabel('Y Label')  
ax.set_zlabel('Z Label')  
  
# 显示图形  
plt.show()

二 解析

以上代码的核心就是

surf = ax.plot_surface(X, Y, Z, cmap='viridis')  

十分简洁明了，希望同学们喜欢