详解numpy.random.shuffle函数

numpy的random模块中的shuffle函数用于np.ndarray数组中的元素打乱顺序，进打乱多维数组的第一维顺序。本博客详细节将该函数的API，并给出示例代码和结果。

函数原型

random.shuffle(x)

参数解析

• x：np.ndarray或者其他可以改变顺序的序列数据。
• 无返回值。

该函数的注意事项

• 直接打乱原数据，元数据会被修改。
• 多维数组只会在其第一维度上面进行顺序打乱，其他维度不发生变化。

例子

示例代码

import numpy as np
np.random.seed(seed=0)

a = np.array(
    [
        [1, 2, 3],
        [2, 1, 3],
        [4, 0, 1],
    ], dtype=np.int_
)

print(f"原始数组为: \n{a}")
np.random.shuffle(a)
print(f"对a使用shuffle函数, 仅沿着a的第一个维度(a的行方向)进行重新排序, 排序后为: \n{a}")
for rol in a:
    np.random.shuffle(rol)
print(f"要相对a的每一行也进行重新排序, 对a的每一行都使用shuffle函数即可, 排序后为: \n{a}")

示例结果

参考

1. numpy.random.choice

收集整理和创作不易, 若有帮助