mgrid 的参数可以是数组

是的，`np.mgrid` 的参数可以是数组，不一定是切片对象。当参数是数组时，它们会被自动转换为切片对象。

例如，`np.mgrid[0:3, 0:2]` 等价于 `np.mgrid[slice(0, 3), slice(0, 2)]`。同样地，`np.mgrid[np.array([0, 1, 2]), np.array([0, 1])]` 也等价于 `np.mgrid[0:3, 0:2]`。

示例代码：

import numpy as np

x, y = np.mgrid[np.array([0, 1, 2]), np.array([0, 1])]

print(x)
print(y)

输出：

array([[0, 0],
       [1, 1],
       [2, 2]])

array([[0, 1],
       [0, 1],
       [0, 1]])

以上代码中，`np.mgrid[np.array([0, 1, 2]), np.array([0, 1])]` 生成了一个 3x2 的网格点坐标数组。第一行对应 `x` 坐标，第二行对应 `y` 坐标。