Python numpy.mgrid函数方法的使用

numpy.mgrid 是 NumPy 提供的一个函数，用于生成多维网格。它特别适用于创建在特定范围内等间距的多维坐标数组。其使用方法与 Matplotlib 的 meshgrid 类似，但更为强大。

以下是 numpy.mgrid 的基本用法和示例：

基本用法
numpy.mgrid 的基本语法如下：

numpy.mgrid[start1:stop1:step1, start2:stop2:step2, ..., startN:stopN:stepN]
其中 start 是开始值，stop 是结束值（不包含），step 是步长。步长可以是复数，表示生成的点数。

示例
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)
参考文档：Python numpy.mgrid函数方法的使用-CJavaPy