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')