Numpy 配列から特定の列を取得する方法 (例あり)

次の構文を使用して、NumPy 配列から特定の列を取得できます。

 #get column in index position 2 from NumPy array
my_array[:, 2]

次の例は、この構文を実際に使用する方法を示しています。

例 1: NumPy 配列から列を取得する

次のコードは、NumPy 配列から特定の列を取得する方法を示しています。

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
data

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get column in index position 2
data[:, 2]

array([3,7,11])

NumPy 配列から列を取得し、それを列ベクトルとして取得する場合は、次の構文を使用できます。

 #get column in index position 2 (as a column vector)
data[:, [2]]

array([[ 3],
       [7],
       [11]])

例 2: NumPy 配列から複数の列を取得する

次のコードは、NumPy 配列から複数の列を取得する方法を示しています。

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
data

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get columns in index positions 1 and 3 from NumPy array
data[:, [1,3]]

array([[ 2, 4],
       [6, 8],
       [10, 12]])

例 3: NumPy 配列から範囲内の列を取得する

次のコードは、NumPy 配列から範囲内の列を取得する方法を示しています。

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
data

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get columns in index positions 0 through 3 (not including 3)
data[:, 0:3]

array([[ 1, 2, 3],
       [5, 6, 7],
       [9, 10, 11]])

範囲内の最後の値 (この場合は 3) は、返される列範囲に含まれないことに注意してください。

追加リソース

次のチュートリアルでは、NumPy で他の一般的な操作を実行する方法について説明します。

関数を NumPy 配列にマップする方法
NumPy配列に列を追加する方法