使用python绘制二元函数图像的实例

#coding:utf-8import numpy as npimport matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3Ddef function_2(x,y):   #  这里的函数可以任意定义   return np.sum(x**2)   fig = plt.figure()ax = Axes3D(fig)x = np.arange(-3,-3,0.1)y = np.arange(-3,-3,0.1)X,Y = np.meshgrid(x,y)#创建网格，这个是关键Z = function_2(X,Y)plt.xlabel('x')plt.ylabel('y')ax.plot_surface(X,Y,Z,rstride=1,cstride=1,cmap='rainbow')plt.show()