python - 打乱一个 numpy 数组

Question

我有一个二维 numpy 数组，我想将其打乱。最好的方法是将其 reshape 为 1-d、洗牌并再次 reshape 为 2-d，还是可以洗牌而不 reshape ？

仅使用 random.shuffle 不会产生预期的结果，而 numpy.random.shuffle 只会随机播放行:

import random
import numpy as np
a=np.arange(9).reshape((3,3))
random.shuffle(a)
print a

[[0 1 2]
 [3 4 5]
 [3 4 5]]

a=np.arange(9).reshape((3,3))
np.random.shuffle(a)
print a

[[6 7 8]
 [3 4 5]
 [0 1 2]]

Answer 1

您可以告诉 np.random.shuffle 对扁平化版本执行操作:

>>> a = np.arange(9).reshape((3,3))
>>> a
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> np.random.shuffle(a.flat)
>>> a
array([[3, 5, 8],
       [7, 6, 2],
       [1, 4, 0]])