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

Question

给定:

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

test[i] 给出 ith<​​/em> 行(例如 [1, 2])。如何访问 ith<​​/em> 列？ (例如 [1, 3, 5])。另外，这会是一项昂贵的操作吗？

Answer 1

要访问第 0 列:

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

要访问第 0 行:

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

这在 NumPy reference 的第 1.4 节(索引)中有介绍。 .这很快，至少在我的经验中。这肯定比在循环中访问每个元素要快得多。