python - 如何提取矩阵的非连续行和列？

Question

如何从矩阵中提取不连续的行和列。例如，在这个矩阵中，我如何提取第 1、2 和 4 行以及第 1、2 和 4 列？

import numpy as np

a = np.matrix([[1, 2, 3, 4, 5],
               [6, 7, 8, 9, 10],
               [11, 12, 13, 14, 15],
               [16, 17, 18, 19, 20],
               [18, 19, 20, 21, 22]])

所以新矩阵应该是:

b = ([[7, 8 , 10],
     [12, 13, 15],
     [19, 20, 22]])

Answer 1

在doc section linked by hpaulj ，请参见以 从 4x3 数组开始的示例，应使用高级索引选择角元素。

具体是开始的段落这个广播也可以使用函数ix_来实现:

在您的情况下，行是 [1, 2, 4] 并且列相同，所以

rows = np.array([1, 2, 4], dtype=np.intp)
columns = np.array([1, 2, 4], dtype=np.intp)
b = a[np.ix_(rows, columns)]