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 で他の一般的な操作を実行する方法について説明します。