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

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]

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

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

我的问题是：有一种简单的方法可以在一行中完成吗？

Answer 1

使用索引和切片：

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