访问 NumPy 多维数组中的列

NumPy 多维数组提供了一种方便有效的方式来存储和操作数据。使用这些数组时，了解如何访问特定元素（包括列）非常重要。

访问列

给定一个如下所示的 NumPy 数组：

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

要访问此数组的第 i 列，请使用以下语法：

test[:, i]

例如，要访问第一列，您可以编写：

test[:, 0]

这将返回数组 [1, 3, 5].

行与列访问

请注意，上述语法中的第一个索引 (test[:]) 访问行，而第二个索引 (i) 访问列。要访问第 i 行，您可以使用以下语法：

test[i, :]

性能

访问 NumPy 数组中的列相对较快手术。它比循环中迭代单个元素要高效得多。确切的性能取决于数组的大小和结构。