Simplfy行和列提取,numpy

Question

我希望使用单个"花式"切片从矩阵中提取行和列,这可能吗？

m = matrix([[1, 2, 3],
            [4, 5, 6],
            [7, 8, 9]])

我的目标是

matrix([[1, 3],
        [7, 9]])

我有一份我想要的物品清单

d = [0,2]

我可以实现功能

m[d][:,d]

但是有一个更简单的表达方式吗？

Answer 1

你可以这样做numpy.ix_:

m = matrix([[1, 2, 3],
            [4, 5, 6],
            [7, 8, 9]])

d = [0,2]
print m[ix_(d,d)]

会发出:

[[1 3]
 [7 9]]