你可以使用os模块来获取文件夹下的所有文件,使用random模块来打乱文件顺序。下面是一个示例代码:

import os
import random

# 获取文件夹下的所有文件
folder_path = "your_folder_path"
files = os.listdir(folder_path)

# 统计文件数量
file_count = len(files)

# 打乱文件顺序
random.shuffle(files)

# 计算文件分割的索引
train_end = int(file_count * 0.8)
val_end = int(file_count * 0.9)

# 分割文件列表
train_files = files[:train_end]
val_files = files[train_end:val_end]
test_files = files[val_end:]

# 输出文件数量
print("文件数量:", file_count)
print("训练文件数量:", len(train_files))
print("验证文件数量:", len(val_files))
print("测试文件数量:", len(test_files))

# 存储文件名的变量
train_file_names = train_files
val_file_names = val_files
test_file_names = test_files

# 打印文件名
print("训练文件名:", train_file_names)
print("验证文件名:", val_file_names)
print("测试文件名:", test_file_names)

请将your_folder_path替换为你想要操作的文件夹路径。这段代码会输出文件数量以及三个变量中的文件名。

python如何换取某一文件夹下所有文件并统计文件数量以及将文件按照8:1:1的比例进行打乱并将文件名存入三个变量中

原文地址: http://www.cveoy.top/t/topic/iyV6 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录