使用matplotlib绘制3D函数图像

学习并尝试了利用matplotlib进行3D函数图像的绘制

 import matplotlib.pyplot as plt  # 绘图用的模块
 from mpl_toolkits.mplot3d import Axes3D  # 绘制3D坐标的函数
 import numpy as np
 
 def fun(x, y):
     return np.power(x, 2) + np.power(y, 2)
 
 fig1 = plt.figure()  # 创建一个绘图对象
 ax = Axes3D(fig1)  # 用这个绘图对象创建一个Axes对象(有3D坐标)
 X, Y = np.mgrid[-2:2:40j, -2:2:40j]  # 从-2到2分别生成40个取样坐标，并作满射联合
 Z = fun(X, Y)  # 用取样点横纵坐标去求取样点Z坐标
 plt.title("This is main title")  # 总标题
 ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=plt.cm.coolwarm, alpha=0.5)  # 用取样点(x,y,z)去构建曲面
 ax.set_xlabel('x label', color='r')
 ax.set_ylabel('y label', color='g')
 ax.set_zlabel('z label', color='b')  # 给三个坐标轴注明
 plt.show()  # 显示模块中的所有绘图对象

具体输出结果如下：