Python 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值包括端点值。

例子

1）一维网格

import numpy as np

# 生成从0到4（不包括4），步长为1的一维网格
grid = np.mgrid[0:4:1]
print(grid)

2）二维网格

import numpy as np

# 生成二维网格
x, y = np.mgrid[0:3:1, 0:3:1]
print(x)
print(y)

3）使用复数表示步长

如果步长是复数，则表示生成的点数。例如，5j 表示在该范围内生成 5 个点。

import numpy as np

# 生成从0到1的范围内，包含5个点的一维网格
grid = np.mgrid[0:1:5j]
print(grid)

4）三维网格

import numpy as np

# 生成三维网格
x, y, z = np.mgrid[0:3:1, 0:3:1, 0:3:1]
print(x)
print(y)
print(z)

5）使用示例

import numpy as np

# 生成从 0 到 4 的二维网格
x, y = np.mgrid[0:5, 0:5]

print("二维网格 x：")
print(x)

print("二维网格 y：")
print(y)

# 生成从 -1 到 1 的 5 个等间距点
points = np.mgrid[-1:1:5j]

print("一维等间距点：")
print(points)