1. 程式人生 > >np.mgrid函式

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.11:3:0.1

]

表示1:5切片間隔為0.1,1:3切片間隔為0.1

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

表示1:5切片均勻取數,取4個,1:3切片均勻取數,取3個