學習並嘗試了利用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() # 顯示模塊中的所有繪圖對象
具體輸出結果如下：
