Python numpy.mgrid函数方法的使用

手机查看 2020-10-26

NumPy（Numerical Python的缩写）是一个开源的Python科学计算库。使用NumPy，就可以很自然地使用数组和矩阵。NumPy包含很多实用的数学函数，涵盖线性代数运算、傅里叶变换和随机数生成等功能。本文主要介绍一下NumPy中mgrid方法的使用。

numpy.mgrid

numpy.mgrid = <numpy.lib.index_tricks.MGridClass object>

nd_grid实例，它返回密集的多维“meshgrid”。

numpy.lib.index_tricks.nd_grid的实例，该实例在建立索引时返回密集的（或充实的）mesh-grid，以便每个返回的参数具有相同的形状。输出数组的尺寸和数量等于索引尺寸的数量。如果step不是复数，则stop不包括在内。

但是，如果step是一个complex number（例如5j），则其magnitude的整数部分将被解释为指定在start值和stop值之间创建的点数，其中stop值包括端点值。

返回值：

mesh-grid的“ndarrays”都是相同的维度

例子

>>> np.mgrid[0:5,0:5]
array([[[0, 0, 0, 0, 0],
        [1, 1, 1, 1, 1],
        [2, 2, 2, 2, 2],
        [3, 3, 3, 3, 3],
        [4, 4, 4, 4, 4]],
       [[0, 1, 2, 3, 4],
        [0, 1, 2, 3, 4],
        [0, 1, 2, 3, 4],
        [0, 1, 2, 3, 4],
        [0, 1, 2, 3, 4]]])
>>> np.mgrid[-1:1:5j]
array([-1. , -0.5,  0. ,  0.5,  1. ])