python - N 维数组 - Python/Numpy

只是想知道是否有任何聪明的方法来执行以下操作。

我有一个表示 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 是否偶然支持类似的东西？

有任何想法吗？

最佳答案

是的，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])

关于python - N 维数组 - Python/Numpy，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/4780791/