python shuffle怎么实现的_python实现简单的多维shuffle

如何按照shuffle过的idx给元素重新排序

>>> a = np.arange(24).reshape(4,3,2)

>>> idx=numpy.arange(len(a))

>>> print(idx)

[0 1 2 3]

>>> np.random.shuffle(idx)

>>> print(idx)

[0 2 3 1]

>>> a[idx, ...]

array([[[ 0, 1],

[ 2, 3],

[ 4, 5]],

[[12, 13],

[14, 15],

[16, 17]],

[[18, 19],

[20, 21],

[22, 23]],

[[ 6, 7],

[ 8, 9],

[10, 11]]])

>>> print(a.shape)

(4, 3, 2)

>>> a

array([[[ 0, 1],

[ 2, 3],

[ 4, 5]],

[[ 6, 7],

[ 8, 9],

[10, 11]],

[[12, 13],

[14, 15],

[16, 17]],

[[18, 19],

[20, 21],

[22, 23]]])