numpyで特定の行と列のデータを取得する方法は何ですか?

NumPyでは、指定された行や列のデータを取得するためにインデックスを使用することができます。

例えば、2次元のnumpy配列arrがあるとします。この配列から2行目と3列目のデータを取得したい場合は、以下のコードを使用することができます:

import numpy as np

arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

row = 1
column = 2

# 获取指定行
row_data = arr[row, :]

# 获取指定列
column_data = arr[:, column]

print("第2行的数据:", row_data)
print("第3列的数据:", column_data)

結果は次のようになります:

第2行的数据: [4 5 6]
第3列的数据: [3 6 9]

指定された行のデータを取得するためには、arr[row, :]インデックスを使用できます。同様に、指定された列のデータを取得するためには、arr[:, column] インデックスを使用できます。インデックスは0から始まることに注意してください。

bannerAds