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])

Answer 2

您可以使用zip转置表示为列表列表的矩阵：

>>> zip(*grid)[0]
(1, 4, 7)

不仅如此，我还会使用Numpy。

Answer 3

要获取Python中的列，您可以使用：

[row[0] for row in grid]
>>> [1,4,7]

您可以重写代码以获取行

grid[0][:]

因为[:]只复制整个数组，不需要添加索引。

但是，根据你想要实现的目标，我会说最好只写一个小矩阵类来隐藏这个实现的东西。