python - 对二维 numpy 数组中的每一行应用相同的排列

Question

要置换一维数组 A 我知道您可以运行以下代码:

import numpy as np
A = np.random.permutation(A)

我有一个二维数组，想对数组的每一行应用完全相同的排列。有什么方法可以指定 numpy 为你做这件事吗？

Answer 1

为 A 中的列数生成随机排列并索引到 A 的列中，就像这样 -

A[:,np.random.permutation(A.shape[1])]

sample 运行-

In [100]: A
Out[100]: 
array([[3, 5, 7, 4, 7],
       [2, 5, 2, 0, 3],
       [1, 4, 3, 8, 8]])

In [101]: A[:,np.random.permutation(A.shape[1])]
Out[101]: 
array([[7, 5, 7, 4, 3],
       [3, 5, 2, 0, 2],
       [8, 4, 3, 8, 1]])