只是想知道是否有任何巧妙的方法来执行以下操作。
我有一个代表 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/