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