numpyで特定の行と列を取得するにはどうすればいいですか?

numpyで特定の行と列を取得するには、スライス操作を使用できます。

指定された行と列を取得する例を次に示す:

import numpy as np

# 创建一个3x3的矩阵
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

# 获取第一行
row_1 = matrix[0]
print(row_1)  # 输出: [1 2 3]

# 获取第一列
column_1 = matrix[:, 0]
print(column_1)  # 输出: [1 4 7]

# 获取第二行和第三行
row_2_3 = matrix[1:3]
print(row_2_3)  # 输出: [[4 5 6]
                #        [7 8 9]]

# 获取第二列和第三列
column_2_3 = matrix[:, 1:3]
print(column_2_3)  # 输出: [[2 3]
                   #        [5 6]
                   #        [8 9]]

NumPyでのスライス操作では、範囲を表すためにコロン(:)が使われ、行と列の範囲を区切るためにコンマ(,)が使用されます。

bannerAds