Python NumPy Random Generator shuffle方法用法及代码示例

例子

基本用法

考虑以下 NumPy 数组：

import numpy as np
x = np.arange(10)
x



array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

要打乱此 NumPy 数组：

rng = np.random.default_rng()
rng.shuffle(x)
x



array([4, 0, 2, 9, 6, 3, 1, 5, 8, 7])

请注意Shuffle[洗牌]是如何就地完成的。

设定轴

考虑以下二维数组：

x = np.arange(12).reshape((3,4))
x



array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

默认情况下， axis=0 ，这意味着行被打乱：

rng = np.random.default_rng(seed=42)
rng.shuffle(x)   # axis=0
x



array([[ 8,  9, 10, 11],
       [ 4,  5,  6,  7],
       [ 0,  1,  2,  3]])

要打乱列，请设置 axis=1 ：

rng.shuffle(x, axis=1)
x



array([[ 2,  0,  3,  1],
       [ 6,  4,  7,  5],
       [10,  8, 11,  9]])

设置种子

为了能够重现随机性，请像这样设置种子：

rng = np.random.default_rng(seed=42)
x = np.arange(10)
rng.shuffle(x)
x



array([5, 6, 0, 7, 3, 2, 4, 9, 1, 8])