Python 从现有矩阵的行列表创建新的numpy矩阵

我有一个2D numpy数组

a

，还有一个列表

x

。

x

的元素是

A

行的索引。我想创建一个新的矩阵

B

，方法是取

a

的行，如

x

所示。如何执行此操作？

在索引

A

以创建新矩阵

B

时，您可以将

x

作为参数传递，如下所示。看

import numpy as np

A = np.arange(25).reshape((5,5))

x = [1, 2, 4]

B = A[x]

print(B)
# [[ 5  6  7  8  9]
# [10 11 12 13 14]
# [20 21 22 23 24]]