如何從 NumPy 中的多維數組中提取特定列？

從NumPy 中的多維數組中檢索列

在NumPy 中，基本操作是存取多維資料組中的單一元素或資料數組集。雖然使用語法 test[i] 檢索行很簡單，但提取特定列可能略有不同。

要存取 NumPy 陣列的第 i 列，您可以使用語法 test[:, i]。這將傳回一個一維數組，其中包含原始矩陣第 i 列的值。

例如，考慮以下陣列：

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

要取得第一列（包含值1、3 和5），您可以使用：

>>> test[:, 0]
array([1, 3, 5])

同樣，要檢索第二列（包含值2、4 和6），您可以使用：

>>> test[:, 1]
array([2, 4, 6])

請注意，此操作的計算成本並不是特別昂貴，並且是NumPy 中處理多維資料的常用技術。

以上是如何從 NumPy 中的多維數組中提取特定列？的詳細內容。更多資訊請關注PHP中文網其他相關文章！