Question

我正在使用random.shuffle来重新排列2D numpy数组。我遇到了以下问题：

import numpy as np
from random import shuffle as sf 

b = np.array([1,2,3,4,5])
print b
# [1 2 3 4 5]
sf(b)
print b
# [1 4 5 3 2]

a = np.array([[1,2,3],[4,5,6],[7,8,9]])
print a
# [[1 2 3]
#  [4 5 6]
#  [7 8 9]]
sf(a)
print a
# [[1 2 3]
#  [4 5 6]
#  [1 2 3]]

结果显示，当洗牌1D阵列时，一切都是正确的。但是在改组2D阵列时，结果变得奇怪。

为什么抛弃原始数组的第三行，第一行重复两次？

我知道可以有解决方案来解决这个问题，例如首先洗牌指示行ID的1D数组，然后按照混洗ID的顺序提取2D数组。但我确实想明确random.shuffle的实现会发生什么，或者我的代码有什么问题。

Answer 1

来自random模块的shuffle不会处理numpy数组，因为它与嵌套的python列表不完全相同。您应该使用numpy.random模块的shuffle代替。

import numpy as np
from numpy.random import shuffle

arr = np.array([[1,2,3],[4,5,6],[7,8,9]])
shuffle(arr)
print(arr)
# output:
# [[4 5 6]
# [1 2 3]
# [7 8 9]]

python：为什么random.shuffle会更改数组

1 个答案: