np.mgrid的用法

功能：返回多維結構，常見的如2D圖形，3D圖形

np.mgrid[ 第1維，第2維 ，第3維 ， …] 

第n維的書寫形式為：

a:b:c

c表示步長，為實數表示間隔；該為長度為[a,b),左開右閉

或：

a:b:cj

cj表示步長，為復數表示點數；該長度為[a,b]，左閉右閉

 

舉例說明：

1）生成1D數組：

a=np.mgrid[-4:4:3j]
a

在[-4,4]區間內取3個值

返回：

array([-4.,  0.,  4.])

 

2）生成個2D矩陣：

mgrid[[1:3:3j, 4:5:2j]]

生成的是3*2的矩陣

import numpy as np
x, y = np.mgrid[1:3:3j, 4:5:2j]
x

x返回：

array([[1., 1.],
       [2., 2.],
       [3., 3.]])

輸出y:

array([[4., 5.],
       [4., 5.],
       [4., 5.]])

所以表示的結果是：

[[(1,4),(1,5)]
 [(2,4),(2,5)]
 [(3,4),(3,5)]
]

結果值先y向右擴展，再x向下擴展

 

3）生成3D立方體

b = np.mgrid[-1:1:2j,-2:2:2j,-3:3:5j]
b

返回：

array([[[[-1. , -1. , -1. , -1. , -1. ],
         [-1. , -1. , -1. , -1. , -1. ]],

        [[ 1. ,  1. ,  1. ,  1. ,  1. ],
         [ 1. ,  1. ,  1. ,  1. ,  1. ]]],


       [[[-2. , -2. , -2. , -2. , -2. ],
         [ 2. ,  2. ,  2. ,  2. ,  2. ]],

        [[-2. , -2. , -2. , -2. , -2. ],
         [ 2. ,  2. ,  2. ,  2. ,  2. ]]],


       [[[-3. , -1.5,  0. ,  1.5,  3. ],
         [-3. , -1.5,  0. ,  1.5,  3. ]],

        [[-3. , -1.5,  0. ,  1.5,  3. ],
         [-3. , -1.5,  0. ,  1.5,  3. ]]]])