例如:删除后缀为 .bmp 类型的文件
import sys, csv, operator
import os
import glob
# 删除指定类型的文件
if __name__ == '__main__':
loadpath = 'K:\\LIDC-IDIR\\Linshi\\' # 数据集分根目录
for indexA in os.listdir(loadpath): #os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表
# indexA 为数据集根目录下的第一层目录
for indexB in os.listdir(os.path.join(loadpath, indexA)):
if indexB[-3:] == 'pkl': continue # 如果是 .pkl 文件则跳过
for indexC in os.listdir(os.path.join(loadpath, indexA, indexB)):
for indexD in os.listdir(os.path.join(loadpath, indexA, indexB, indexC)): # 到此应该进入到数据集的最后一层
for infile in glob.glob(os.path.join(loadpath, indexA, indexB, indexC,'*.bmp')):
os.remove(infile)