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