Python numpy mgrid用法及代码示例

本文简要介绍 python 语言中 numpy.mgrid 的用法。

用法:

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

nd_grid 实例返回一个密集的多维“meshgrid”。

numpy.lib.index_tricks.nd_grid 的一个实例，它在索引时返回密集(或充实)mesh-grid，以便每个返回的参数具有相同的形状。输出数组的维度和数量等于索引维度的数量。如果步长不是复数，则停止不包括在内。

但是，如果步长是复数(例如 5j)，则其幅度的整数部分被解释为指定要在开始值和结束值之间创建的点数，其中包括结束值。

返回：

mesh-gridndarray所有相同的尺寸

例子：

>>> 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. ])