Python实现高斯函数的三维显示方法

2020-01-04 13:39:57

字体：大中小

来源：转载

供稿：网友

在网上查阅资料，发现很少用Python进行高斯函数的三维显示绘图的，原因可能是其图形显示太过怪异，没有MATLAB精细和直观。

回顾一下二维高斯公式：

Python,高斯函数,三维显示

σ此处取3。

在MATLAB下的程序为：

u=[-10:0.1:10];v=[-10:0.1:10];[U,V]=meshgrid(u,v);H=exp(-(U.^2+V.^2)./2/3^2);mesh(u,v,H); %绘制三维曲面的函数title('高斯函数曲面');

其显示结果为：

Python,高斯函数,三维显示

放大效果显示，很平滑和直观。

Python,高斯函数,三维显示

以下为Python的程序部分：

import numpy as npimport matplotlib.pyplot as pltimport mathimport mpl_toolkits.mplot3dx, y = np.mgrid[-2:2:200j, -2:2:200j]z=(1/2*math.pi*3**2)*np.exp(-(x**2+y**2)/2*3**2)ax = plt.subplot(111, projection='3d')ax.plot_surface(x, y, z, rstride=1, cstride=1, cmap='rainbow', alpha=0.9)#绘面ax.set_xlabel('x')ax.set_ylabel('y')ax.set_zlabel('z')plt.show()

显示效果：

Python,高斯函数,三维显示