python - 使用 Numpy 随机排列数组的列

Question

假设我有一个维度为 (n, m) 的数组 r。我想打乱那个数组的列。

如果我使用 numpy.random.shuffle(r) 它会打乱行。我怎样才能只洗牌？因此第一列随机变为第二列，第三列随机变为第一列，依此类推。

示例:

输入:

array([[  1,  20, 100],
       [  2,  31, 401],
       [  8,  11, 108]])

输出:

array([[  20, 1, 100],
       [  31, 2, 401],
       [  11,  8, 108]])

Answer 1

一种方法是打乱转置数组:

 np.random.shuffle(np.transpose(r))

另一种方法(参见 YXD 的回答 https://stackoverflow.com/a/20546567/1787973 )是生成排列列表以按该顺序检索列:

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

在性能方面，第二种方法更快。