python取矩阵某些列

在 Python 中，要取矩阵的某些列，可以使用 NumPy 库的切片功能。

假设有一个二维矩阵 mat，要取第 2, 3, 5 列，可以使用以下代码：

import numpy as np

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

# 取第 2, 3, 5 列
cols = [1, 2, 4]
res = mat[:, cols]

print(res)

# 输出结果为：
# array([[ 2,  3,  5],
#        [ 7,  8, 10],
#        [12, 13, 15]])

其中，`:` 表示取所有的行，`cols` 列表中存储了要取的列的下标，使用`[:, cols]`即可取出矩阵的这些列。

需要注意的是，`cols` 列表中的下标从 0 开始计数，因此第 2, 3, 5 列的下标分别是 1, 2, 4。