N维数组 - Python/Numpy

Question

只是想知道是否有任何聪明的方法来做以下事情.

我有一个代表3x3网格的N维数组

grid = [[1,2,3],
        [4,5,6],
        [7,8,9]]

为了获得第一行,我执行以下操作:

grid[0][0:3]
>> [1,2,3]

为了获得第一列,我想做这样的事情(尽管不可能):

grid[0:3][0]
>> [1,4,7]

NumPy是否偶然支持与此相似的任何内容？

---

有任何想法吗？

Answer 1

是的,在Numpy中有类似的东西:

import numpy as np

grid = np.array([[1, 2, 3],
                 [4, 5, 6],
                 [7, 8, 9]])

grid[0,:]
# array([1, 2, 3])

grid[:,0]
# array([1, 4, 7])