给定具有排列的列表，重新排列给定 numpy 2D 数组的行

Question

有一种非常简单的方法可以像这样用 numpy 置换 2d 数组的 列：

array1 = np.array([[11, 22, 33, 44, 55],
                  [66,  77,  88,  99, 100]])

print("Original array:")
print(array1)
permutation = [1,3,0,4,2]

result = array1[:, permutation]
print("New array:")
print(result)

这个输出：

Original array:
[[ 11  22  33  44  55]
 [ 66  77  88  99 100]]
New array:
[[ 22  44  11  55  33]
 [ 77  99  66 100  88]]

Visual representation (from w3resource.com)

有没有一种方法可以优雅地完成相同的事情，但只针对 行？

Answer 1

正如@Marat 在 cmets 中提到的，您可以通过您为列描述的类似高级索引来执行相同操作：

array1 = np.array([[11, 22, 33, 44, 55],
                  [66,  77,  88,  99, 100]])
permutation = [1,0]
array1[permutation]
#[[ 66  77  88  99 100]
# [ 11  22  33  44  55]]

当你提前索引numpy数组时，默认是调用rows。