从指定的文件夹中随机抽取指定数量的文件，并将它们复制到当前目录下的‘random_files‘文件夹中。

最新推荐文章于 2025-05-10 17:09:06 发布

原创最新推荐文章于 2025-05-10 17:09:06 发布 · 691 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python

该代码定义了一个函数，用于从指定文件夹中按给定数量随机选择文件，并将这些文件复制到当前目录的random_files文件夹。如果提供的随机种子不为空，则会设置随机数生成器的种子。在示例中，它展示了如何使用这个函数来抽取10个文件，并使用种子123确保可重复性。

import os
import random
import shutil

def random_select_files(folder_path, num_files, seed=None):
    """
    从指定文件夹中随机抽取指定数量的文件，并将其复制到当前目录下的'random_files'文件夹中
    :param folder_path: 文件夹路径
    :param num_files: 需要抽取的文件数量
    :param seed: 随机种子
    """
    if not os.path.isdir(folder_path):
        print(f"{folder_path}不是一个有效的文件夹路径")
        return
    files = os.listdir(folder_path)
    if num_files > len(files):
        print("文件数量不足，无法完成抽取")
        return
    if seed:
        random.seed(seed)
    random.shuffle(files)
    random_files = files[:num_files]
    if not os.path.exists('random_files'):
        os.mkdir('random_files')
    for file in random_files:
        shutil.copy(os.path.join(folder_path, file), 'random_files')
    print(f"{num_files}个文件已被成功抽取到'random_files'文件夹中")

# 示例
folder_path = 'path/to/folder' # 替换为实际文件夹路径
num_files = 10 # 需要抽取的文件数量
seed = 123 # 随机种子
random_select_files(folder_path, num_files, seed)