python删除一个目录下某个类型的文件，以及删除目录下子目录的所有文件

import os
for files in os.listdir(‘output‘):
        if files.endswith(".py"):
            os.remove(os.path.join(‘output‘,files))

import os,shutil

def del_file(filepath):
    """
    删除某一目录下的所有文件或文件夹
    :param filepath: 路径
    :return:
    """
    del_list = os.listdir(filepath)
    for f in del_list:
        file_path = os.path.join(filepath, f)
        if os.path.isfile(file_path):
            os.remove(file_path)
        elif os.path.isdir(file_path):
            shutil.rmtree(file_path)
del_file(‘test_fold‘)