Python大數(shù)據(jù)培訓：繪制3D曲面圖

　　Python可視化技術中，經(jīng)常會運用到3D曲面圖，今天我們就來講一下如何實現(xiàn)3D曲面圖。Axes3D類的對象使用plot_surface()方法繪制3D曲面圖，該方法的語法格式如下所示：

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

　　該方法常用參數(shù)含義如下。

　　.X，Y，Z：表示x、y、z軸的數(shù)據(jù)。

　　.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曲面圖