我有一个 2D numpy 数组 A 和一个列表 x。 x 的元素是 A 行的索引。我想通过采用 A 的行(如 x 所示)来创建一个新的矩阵 B。我怎样才能做到这一点？
最佳答案
您可以在索引 A 时将 x 作为参数传递，以创建新矩阵 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]]
关于python - 从现有矩阵的行列表创建新的 numpy 矩阵，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/25311776/