Python大数据培训：绘制3D曲面图

　　Python可视化技术中，经常会运用到3D曲面图，今天我们就来讲一下如何实现3D曲面图。Axes3D类的对象使用plot_surface()方法绘制3D曲面图，该方法的语法格式如下所示：

plot_surface(self, X, Y, Z, *args, norm=None, vmin=None, vmax=None,
             lightsource=None, **kwargs)

　　该方法常用参数含义如下。

　　.X，Y，Z：表示x、y、z轴的数据。

　　.rcount，ccount：表示每个坐标轴方向所使用的最大样本量，默认为50。

　　.rstride，cstride：表示采样的密度。

　　.color：表示曲面的颜色

　　.cmap：表示曲面的颜色映射表。

　　.shade：表示是否对曲面进行着色。

　　绘制3D曲面图的示例代码如下。

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
from matplotlib import cm
import numpy as np
x1 = np.arange(-5, 5, 0.25)
y1 = np.arange(-5, 5, 0.25)
x1, y1 = np.meshgrid(x1, y1)
r1 = np.sqrt(x1**2 + y1**2)
z1 = np.sin(r1)
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 绘制曲面图
ax.plot_surface(x1, y1, z1, cmap=cm.coolwarm, linewidth=0,
                antialiased=False)
# 设置z轴刻度的范围、位置、格式
ax.set_zlim(-1.01, 1.01)
plt.show()

　　运行程序，效果如图7-2所示。

　　图7-2 3D曲面图