将N张图片保存在多个文件夹中(Python)

我有多个文件夹,其中每个文件夹包含多个图像。我的目标是编写一个python脚本,以随机地在每个文件夹中保留N张图像。

如果有人可以帮助我,我将不胜感激。谢谢。

评论
cnisi
cnisi

如果您有目录列表,这可能会对您有所帮助:

import os
import random
def keep_n_dir(directory, n):
    files = os.listdir(directory) #You retrieve the list of names of files
    if len(files) > n: #If you already have less than n files, you do nothing
        diff = len(files) - n
        files_to_delete = random.sample(files, k=diff) 
        for file in files_to_delete: 
            os.remove(os.path.join(directory, file))
点赞
评论