如何访问 NumPy 多维数组中的特定列？

访问 NumPy 多维数组中的列

给定一个多维 NumPy 数组，使用索引运算符 test[i] 按行访问其元素是直截了当。但是，提取特定列可能会令人困惑。

列索引

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

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

例如，给定数组：

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

您可以按如下方式访问第一列：

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

性能注意事项

此列索引操作通常是高效的。它直接访问与所需列相对应的内存位置，避免了 for 循环的开销。然而，实际性能可能会因数组大小和内存布局等因素而有所不同。

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