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

小能豆

python

鉴于：

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

test[i]给出第 i行（例如[1, 2]）。我如何访问第 i列？（例如[1, 3, 5]）。此外，这会是一个昂贵的操作吗？

阅读 31

2024-09-15

共1个答案

小能豆

和：

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

要访问第 0 列：

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

要访问第 0 行：

>>> test[0, :]
array([1, 2])

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

2024-09-15