numpy - Python - numpy mgrid 和 reshape

Question

有人可以向我解释这段代码的第二行是做什么的吗？

objp = np.zeros((48,3), np.float32)
objp[:,:2] = np.mgrid[0:8,0:6].T.reshape(-1,2)

有人可以向我解释代码的 np.mgrid[0:8,0:6] 部分到底在做什么以及代码的 T.reshape(-1,2) 部分到底在做什么吗？

谢谢，干得好!

Answer 1

查看这些的最简单方法是对 mgrid 使用较小的值。 :

In [11]: np.mgrid[0:2,0:3]
Out[11]:
array([[[0, 0, 0],
        [1, 1, 1]],

       [[0, 1, 2],
        [0, 1, 2]]])

In [12]: np.mgrid[0:2,0:3].T  # (matrix) transpose
Out[12]:
array([[[0, 0],
        [1, 0]],

       [[0, 1],
        [1, 1]],

       [[0, 2],
        [1, 2]]])

In [13]: np.mgrid[0:2,0:3].T.reshape(-1, 2)  # reshape to an Nx2 matrix
Out[13]:
array([[0, 0],
       [1, 0],
       [0, 1],
       [1, 1],
       [0, 2],
       [1, 2]])

然后 objp[:,:2] =设置 objp 的第 0 和第 1 列到这个结果。