NumPy 随机排列

元素的随机排列

排列是指元素的布局。例如，[3, 2, 1] 是 [1, 2, 3] 的一个排列，反之亦然。

NumPy 的 Random 模块为此提供了两个方法：shuffle() 和 permutation()。

数组的洗牌

洗牌是指就地改变元素的排列。即，在数组本身中改变。

实例

随机打乱以下数组的元素：

from numpy import random
import numpy as np

arr = np.array([1, 2, 3, 4, 5])

random.shuffle(arr)

print(arr)

亲自试一试

提示：shuffle() 方法会更改原始数组。

生成数组的排列

实例

生成以下数组元素的随机排列：

from numpy import random
import numpy as np

arr = np.array([1, 2, 3, 4, 5])

print(random.permutation(arr))

亲自试一试

提示：permutation() 方法返回重新排列的数组（并保持原始数组不变）。