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])