如何访问 NumPy 多维数组中的各个列？

访问 NumPy 多维数组中的各个列

NumPy 提供了用于处理多维数组的强大工具。通常，有必要访问这些数组的各个列。本指南将演示如何轻松实现此目的。

想象以下 NumPy 数组：

<code class="python">test = np.array([[1, 2], [3, 4], [5, 6]])</code>

要访问该数组的第 i 行，只需使用 test[i]。但是，如果您需要访问第 i 列，则语法略有不同。

要访问第 i 列，请使用以下语法：

<code class="python">test[:, i]</code>

例如，要访问第一个列列：

<code class="python">>>> test[:, 0]
array([1, 3, 5])</code>

请注意，第一个索引 (:) 选择所有行，而第二个索引 (0) 指定列。

此功能记录在 NumPy 的第 1.4 节中参考。就性能而言，以这种方式访问​​列通常非常高效，并且比在循环中单独访问每个元素要快得多。

以上是如何访问 NumPy 多维数组中的各个列？的详细内容。更多信息请关注PHP中文网其他相关文章！