使用matplotlib繪製3D函數圖像

學習並嘗試了利用matplotlib進行3D函數圖像的繪製

 1 import matplotlib.pyplot as plt  # 繪圖用的模塊
 2 from mpl_toolkits.mplot3d import Axes3D  # 繪製3D坐標的函數
 3 import numpy as np
 4 
 5 
 6 def fun(x, y):
 7     return np.power(x, 2) + np.power(y, 2)
 8 
 9 
10 fig1 = plt.figure()  # 創建一個繪圖對象
11 ax = Axes3D(fig1)  # 用這個繪圖對象創建一個Axes對象(有3D坐標)
12 X, Y = np.mgrid[-2:2:40j, -2:2:40j]  # 從-2到2分別生成40個取樣坐標，並作滿射聯合
13 Z = fun(X, Y)  # 用取樣點橫縱坐標去求取樣點Z坐標
14 plt.title("This is main title")  # 總標題
15 ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=plt.cm.coolwarm, alpha=0.5)  # 用取樣點(x,y,z)去構建曲面
16 ax.set_xlabel('x label', color='r')
17 ax.set_ylabel('y label', color='g')
18 ax.set_zlabel('z label', color='b')  # 給三個坐標軸註明
19 plt.show()  # 顯示模塊中的所有繪圖對象

具體輸出結果如下：