python选取矩阵中的某几列

可以使用 numpy 库中的切片操作来选取矩阵中的某几列。

假设我们有一个 3 行 5 列的矩阵 mat，我们想要选取第 2 列和第 4 列，可以使用以下代码：

import numpy as np

mat = np.array([[1, 2, 3, 4, 5],
                [6, 7, 8, 9, 10],
                [11, 12, 13, 14, 15]])

selected_cols = mat[:, [1, 3]]

其中，`mat[:, [1, 3]]` 表示选取矩阵 mat 中所有行的第 2 列和第 4 列，返回一个新的矩阵。这里使用了冒号 `:` 表示选取所有行。

如果我们想要选取连续的列，可以使用切片操作：

selected_cols = mat[:, 1:4]  # 选取第 2 列到第 4 列

其中，`1:4` 表示从第 2 列到第 4 列，不包括第 4 列。