根据If条件

Question

以下代码计算每个子目录中的图像数量。如果子目录中的图像大于2，如何删除子目录 n13是主目录 =＆gt;其中 300个子目录（1 ... 300） =＆gt;每个子目录都有图像。

输出：
图像：2，目录：1
图像：3，目录：2
图片：4，目录：3

import os
path='C:/n13/'
def count_em(path):
    x = 0
    for root, dirs, files in os.walk(path):
       files_count = (len(files))
       x = x + 1
       print("Images:",files_count,"Directory:",x)
    return files_count

Answer 1

您可以使用shutil.rmtree()删除包含其子目录和文件的文件夹。

import os
import shutil

path='C:/n13/'

def count_em(path):
    x = 0
    files_count = 0
    for root, dirs, files in os.walk(path):
        files_count = (len(files))
        if files_count >= 2:
            shutil.rmtree(root)
        x = x + 1
        print("Images:", files_count, "Directory:", x)
    return files_count


count_em(path)