np.mgrid函数

代码：

z = np.mgrid[1:5, 1:3]
x, y = z[0], z[1]
print(x)
print(y)

结果：

[[1 1]
 [2 2]
 [3 3]
 [4 4]]
[[1 2]
 [1 2]
 [1 2]

 [1 2]]

首先np.mgrid输出至少是一个三维的向量。

其中的元素，z[0],z[1]都是二维矩阵。

z[0]行数由np.mgrid第一个参数决定，上例为1：5，且为1 2 3 4，列数由1：3决定，利用广播机制填充。

z[1]列数由np.mgrid第二个参数决定，上例为1：3，且为1 2 ，行数由1：5决定，利用广播机制填充。

PS：z = np.mgrid[1:5:0.1, 1:3:0.1]

表示1：5切片间隔为0.1，1：3切片间隔为0.1

z = np.mgrid[1:5:4j, 1:3:3j]

表示1：5切片均匀取数，取4个，1：3切片均匀取数，取3个