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

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

 #get row 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
print (data)

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

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

array([ 9, 10, 11, 12])

NumPy 配列のインデックス位置 2 の行のみが返されることに注意してください。

例 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 rows in index positions 0 and 2 from NumPy array
data[[0,2],:]

array([[ 1, 2, 3, 4],
       [9, 10, 11, 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 rows in index positions 0 through 1
data[0:2, :]

array([[1, 2, 3, 4],
       [5, 6, 7, 8]])

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

追加リソース

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

NumPy配列から特定の列を取得する方法
関数を NumPy 配列にマップする方法
NumPy配列に列を追加する方法