Python NumPy Random Generator permutation方法用法及代碼示例

例子

傳遞一個整數

要獲取 [0,1,2,3,4] 的打亂數組：

import numpy as np
rng = np.random.default_rng(seed=42)
rng.permutation(5)



array([4, 2, 3, 1, 0])

請注意，這相當於 rng.permutation(np.arange(5)) 。

傳入一個數組

隨機打亂數字數組：

rng = np.random.default_rng(seed=42)
rng.permutation([5,2,6,1])



array([1, 6, 2, 5])

請注意，打亂一維數組時，行為與 permuted(~) 完全相同。

設定軸

考慮以下二維數組：

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.permutation(x)   # axis=0



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

洗牌列

隨機打亂二維數組的列：

rng = np.random.default_rng(seed=42)
rng.permutation(x, axis=1)



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