python 同时提取矩阵的某行和某列

同样地，你可以使用 NumPy 中的切片操作来同时提取矩阵的某行和某列。以下是一些示例代码：

import numpy as np

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

# 提取第一行和第二列
row_1_col_2 = mat[0, 1]
print(row_1_col_2)  # 输出 2

# 提取第一行和第二行的第二列
rows_1_2_col_2 = mat[:2, 1]
print(rows_1_2_col_2)
# 输出
# [2 5]

# 提取第二列和第三列的第一行
cols_2_3_row_1 = mat[0, 1:]
print(cols_2_3_row_1)
# 输出
# [2 3]

在上面的代码中，`mat` 是一个 3x3 的矩阵。我们使用 `mat[0, 1]` 来提取第一行和第二列，使用 `mat[:2, 1]` 来提取第一行和第二行的第二列，使用 `mat[0, 1:]` 来提取第二列和第三列的第一行。