python 中random.shuffle()函数将序列的所有元素进行随机排序

一、函数详解：

二、示例：

import random

a = [1,2,3,4,5,6]
random.shuffle(a)
print(a)   ## [1, 5, 6, 3, 2, 4]

三、应用：

可以用来划分数据集，如划分为训练集、验证集。

import random
import os

## 超参
in_dir = 'folder1/number.txt'
out_dir = 'folder1'
val_size = 3

## 读数据
with open(in_dir,'r',encoding='utf-8') as f:
    lines = f.readlines()

## 随机排序
random.shuffle(lines)

## 划分训练集、验证集
with open(os.path.join(out_dir, "train.txt"), "w", encoding="utf-8") as f:
    for m in lines[val_size :]:
        f.write(m.strip() + "\n")
with open(os.path.join(out_dir, "val.txt"), "w", encoding="utf-8") as f:
    for m in lines[: val_size]:
        f.write(m.strip() + "\n")