更新時間:2022年05月06日10時39分 來源:傳智教育 瀏覽次數(shù):
Python可視化技術(shù)中,經(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:表示每個坐標(biāo)軸方向所使用的最大樣本量,默認(rèn)為50。
.rstride,cstride:表示采樣的密度。
.color:表示曲面的顏色
.cmap:表示曲面的顏色映射表。
.shade:表示是否對曲面進(jìn)行著色。
繪制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) # 設(shè)置z軸刻度的范圍、位置、格式 ax.set_zlim(-1.01, 1.01) plt.show()
運行程序,效果如圖7-2所示。
圖7-2 3D曲面圖