根据If条件删除当前目录

Question

以下代码计算每个子目录中的图像数量。如果子目录中的图像超过 2 个，如何删除子目录。
n13 是主目录=> 有 300 个子目录(1...300)=> 每个子目录都有图片。

输出：
图片：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)