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

Question

我有一个2D numpy数组A和一个列表x。 x的元素是A行的索引。我想创建一个新的矩阵B，方法是按A所示的x行。我怎么能这样做？

Answer 1

在索引x以创建新矩阵A时，您可以将B作为参数传递，如下所示。请参阅the docs here。

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]]