使用Python随机选择文件

前言

当我们有大量文件需要查看的时候,如果是简单挑选前几个文件可能会有bias。那么如何随机挑选其中一些文件并保存到另外一个文件夹里以供我们查看呢? 本文会以以下600余张图片为例,教大家如何使用python随机挑选100个文件或100个指定类型的文件。


目录

一、随机选择文件示例

 二、随机选择指定后缀文件示例


一、随机选择文件示例

我们可以使用os模块来遍历源文件夹中的所有文件,然后使用random.sample()方法从中随机选择50个文件,并使用shutil模块将它们复制到目标文件夹中。具体程序如下:

import os
import random
import shutil

# 设置源文件夹路径
src_dir_path = "路径/到/源文件夹"

# 设置目标文件夹路径
dst_dir_path = "路径/到/目标文件夹"

# 获取源文件夹中所有文件的列表
files = [os.path.join(src_dir_path, f) for f in os.listdir(src_dir_path)]

# 随机选择100个文件,如果你想更改文件数量,只需要更改这个100即可
random_files = random.sample(files, 100)

# 将选中的文件复制到目标文件夹中
for file_path in random_files:
    # 如果目标文件夹不存在,则创建它
    if not os.path.exists(dst_dir_path):
        os.makedirs(dst_dir_path)

    # 将文件复制到目标文件夹中
    shutil.copy(file_path, dst_dir_path)

选择后的图片:

 二、随机选择指定后缀文件示例

import os
import random
import shutil

# 设置源文件夹路径
src_dir_path = "路径/到/源文件夹"

# 设置目标文件夹路径
dst_dir_path = "路径/到/目标文件夹"

# 获取源文件夹中所有png文件的列表,如果你希望更改文件类型,只需要更改.png即可
txt_files = [os.path.join(src_dir_path, f) for f in os.listdir(src_dir_path) if os.path.splitext(f)[1] == ".png"]

# 随机选择100个png文件
random_txt_files = random.sample(txt_files, 100)

# 将选中的png文件复制到目标文件夹中
for file_path in random_txt_files:
    # 如果目标文件夹不存在,则创建它
    if not os.path.exists(dst_dir_path):
        os.makedirs(dst_dir_path)

    # 将文件复制到目标文件夹中
    shutil.copy(file_path, dst_dir_path)

如果你希望更改文件类型,只需要更改.png即可。

猜你喜欢

转载自blog.csdn.net/weixin_49030835/article/details/129353611