獲取 NumPy 陣列的列

本教程將介紹從 Python 的多維 NumPy 陣列中獲取特定列的方法。

使用基本切片方法從多維 NumPy 陣列獲取列

基本切片方法的工作原理與 Python 中的列表切片相同。我們可以使用基本的切片方法從多維 NumPy 陣列中獲取特定的列。基本切片方法將建立現有陣列的新檢視，而不是建立陣列的新副本。這些新的檢視指向現有的陣列，在所有指向原始陣列的檢視被垃圾收集器收集之前，原始陣列的記憶體不能被釋放。下面的程式碼例子向我們展示瞭如何用 Python 的基本切片方法從一個多維 NumPy 陣列中獲得一個特定的列。

import numpy as np
array = np.array([[1,2],[3,4],[5,6],[7,8],[9,0]])
print(array[:,1])

輸出：

[2 4 6 8 0]

在上面的程式碼中，我們在 Python 中使用 [:,1] 切片索引提取了多維 NumPy 陣列 array 的第二列。索引的第一部分是行的索引。我們將第一部分留空，因為我們要選擇所有行。:運算子表示索引中的選擇操作。我們使用 [:,1] 作為列索引，以獲取每一行的第二列。