如何访问 NumPy 多维数组的第 i 列？

小编典典

all

假设我有：

test = numpy.array([[1, 2], [3, 4], [5, 6]])

test[i]得到数组的 第 i 行（例如[1, 2]）。如何访问第 i 列？（例如[1, 3, 5]）。另外，这会是一项昂贵的操作吗？

阅读 140

2022-03-07

共1个答案

小编典典

>>> test[:,0]
array([1, 3, 5])

相似地，

>>> test[1,:]
array([3, 4])

让您访问行。这在NumPy
参考的第 1.4
节（索引）中有介绍。这很快，至少在我的经验中。这肯定比在循环中访问每个元素要快得多。

2022-03-07