如何存取 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中文網其他相關文章！