python中mgrid和meshgrid的不同

在生成插值数据时，有时会用到 mgrid 和 meshgrid 两个函数。
关于插值函数的应用参见 https://blog.csdn.net/gsgbgxp/article/details/116145501#t7

关于两者的差别可参见

• https://www.jb51.net/article/275907.htm
• https://blog.csdn.net/Whisper321/article/details/93871132

一、meshgrid函数

官方文档：numpy.meshgrid

meshgrid函数通常使用在数据的矢量化上。

它适用于生成网格型数据，可以接受两个一维数组生成两个二维矩阵，对应两个数组中所有的(x,y)对。

示例展示：

由上面的示例展示可以看出，meshgrid的作用是：

根据传入的两个一维数组参数生成两个数组元素的列表。

如果第一个参数是xarray，维度是xdimesion，

第二个参数是yarray，维度是ydimesion。

那么生成的第一个二维数组是以xarray为行，共ydimesion行的向量；

而第二个二维数组是以yarray的转置为列，共xdimesion列的向量。

二、mgrid函数

官方文档 numpy.mgrid

用法：返回多维结构，常见的如2D图形，3D图形。对比np.meshgrid，在处理大数据时速度更快，且能处理多维（np.meshgrid只能处理2维）
ret = np.mgrid[ 第1维，第2维 ，第3维 ， …]
返回多值，以多个矩阵的形式返回，

第1返回值为第1维数据在最终结构中的分布，

第2返回值为第2维数据在最终结构中的分布，以此类推。（分布以矩阵形式呈现）

例如np.mgrid[X , Y]
样本（i，j）的坐标为 （X[i，j] ,Y[i，j]）,X代表第1维，Y代表第2维，在此例中分别为横纵坐标。

例如1D结构（array），如下：

In [2]: import numpy as np   In [3]: pp=np.mgrid[-5:5:5j]   In [4]: pp
Out[4]: array([-5. , -2.5,  0. ,  2.5,  5. ])

例如2D结构 (2D矩阵)，如下：

>>> pp = np.mgrid[-1:1:2j,-2:2:3j]
>>> x , y = pp
>>> x
array([[-1., -1., -1.],
       [ 1.,  1.,  1.]])
>>> y 
array([[-2.,  0.,  2.],
       [-2.,  0.,  2.]])

例如3D结构 (3D立方体)，如下：

>>> pp = np.mgrid[-1:1:2j,-2:2:3j,-3:3:5j]
>>> print pp
[[[[-1.  -1.  -1.  -1.  -1. ]
   [-1.  -1.  -1.  -1.  -1. ]
   [-1.  -1.  -1.  -1.  -1. ]]
 
  [[ 1.   1.   1.   1.   1. ]
   [ 1.   1.   1.   1.   1. ]
   [ 1.   1.   1.   1.   1. ]]]
 
 
 [[[-2.  -2.  -2.  -2.  -2. ]
   [ 0.   0.   0.   0.   0. ]
   [ 2.   2.   2.   2.   2. ]]
 
  [[-2.  -2.  -2.  -2.  -2. ]
   [ 0.   0.   0.   0.   0. ]
   [ 2.   2.   2.   2.   2. ]]]
 
 
 [[[-3.  -1.5  0.   1.5  3. ]
   [-3.  -1.5  0.   1.5  3. ]
   [-3.  -1.5  0.   1.5  3. ]]
 
  [[-3.  -1.5  0.   1.5  3. ]
   [-3.  -1.5  0.   1.5  3. ]
   [-3.  -1.5  0.   1.5  3. ]]]]

三、meshgrid 和 mgrid 的区别

参考链接： https://blog.csdn.net/Whisper321/article/details/93871132

mgrid[[1:3:3j, 4:5:2j]]
3j：3个点

• 步长为复数表示点数，左闭右闭
• 步长为实数表示间隔，左闭右开

更多例子

import numpy as np 
def func(x,y): 
    return (x+y)*np.exp(-5*(x**2+y**2)) 
x,y = np.mgrid[-1:1:2j,-1:1:3j] 
p,q = np.meshgrid(np.linspace(-1,1,2),np.linspace(-1,1,3))
a,b = np.meshgrid(np.linspace(-1,1,3),np.linspace(-1,1,2))
z = func(x,y)
r = func(p,q)
c = func(a,b)

运行结果为