python：批量刪除指定文件目錄中多個文件

#coding:utf-8
# 任務需要，需要刪除多余的文件，手動刪除太麻煩，幾行python搞定
import os
from glob import glob
path = r"/media/icml-014/peixu/MyData/2019brain/valid/*" # 獲取到目錄
all_file = glob(path)  # 得到該目錄下的所有文件
for i in range(len(all_file)):
    filename = all_file[i]
    print("filename:",filename) #獲取子目錄文件名
    sub_file = os.path.join(path, filename)  
    print("sub_file:", sub_file)
    all_sub_file = glob(os.path.join(sub_file,"*"))  # 得到子目錄下所有文件
    for j in range(len(all_sub_file)):
        sub_file_name = all_sub_file[j]
        print("sub_file_name:", sub_file_name)
        name = sub_file_name.split("f32.")[-1] 
        if name =="pkl": # 找到自己要刪出的文件類型，我刪除的文件類型是.pkl文件。
            print("del_sub_file_name:", sub_file_name)
            os.remove(sub_file_name)   # 進行刪除