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