import os
import shutil
# 假设 'directory_A' 是目录A的路径
directory_A = '/home/unitree/newHardDisk/VINT_data_train/go_stanford/'# 遍历目录A中的每个子目录，并计算每个子目录中的文件数量
def count_files_in_subdirectories(directory):# 存储每个子目录文件数量的字典subdirectory_file_counts = {}# 列出给定目录中的所有项for item in os.listdir(directory):item_path = os.path.join(directory, item)# 检查此项是否为目录if os.path.isdir(item_path):# 列出子目录中的所有文件files = [f for f in os.listdir(item_path) if os.path.isfile(os.path.join(item_path, f))]# 更新字典中的文件计数subdirectory_file_counts[item] = len(files)return subdirectory_file_counts# 调用函数并获取文件计数
file_counts = count_files_in_subdirectories(directory_A)# 打印结果
for subdir, count in file_counts.items():print(f"子目录 {subdir} 中的文件数量: {count}")if count<5:print(f"删除子目录 {subdir}")shutil.rmtree(f"{directory_A}/{subdir}")