通过指定行/列，从现有矩阵创建新矩阵

Question

在NumPy中，假设我有一个方形矩阵X:

X = array([[3, 1, 4, 5], [5, 1, 2, 1], [4, 4, 0, 1], [0, 3, 0, 3]])

我现在想要选择这个矩阵的某个行/列集合，并创建一个新的矩阵Y。这些行/列的索引存储在一个列表z中。

在这个例子中，假设我的下标是:

z = [0, 2]

然后我的新矩阵将从X中获取0th和2nd行/列:

Y = array([[3, 4], [4, 0]])

我的问题是:有没有一种简单的方法在一行中完成这个?

Answer 1

使用索引和切片:

>>> X[z][:,z]
array([[3, 4],
       [4, 0]])