python矩阵行列的选取

在Python中，我们可以使用numpy库来对矩阵进行行列的选取。

选取行：

import numpy as np

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

# 选取第1行
row1 = matrix[0]

# 选取第2行
row2 = matrix[1]

print(row1)  # 输出 [1 2 3]
print(row2)  # 输出 [4 5 6]

选取列：

import numpy as np

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

# 选取第1列
col1 = matrix[:, 0]

# 选取第2列
col2 = matrix[:, 1]

# 选取第3列
col3 = matrix[:, 2]

print(col1)  # 输出 [1 4]
print(col2)  # 输出 [2 5]
print(col3)  # 输出 [3 6]

在numpy中，选取行和选取列的方式是不同的。选取行时，我们直接使用矩阵名加索引号即可；而选取列时，需要使用"[:, 列索引号]"的方式来选取。