【功能脚本】删除列表中指定的文件

import os
"""
在磁盘中删除列表中指定文件名的文件
具体功能：list中是想要删除的文件名称，文件在磁盘中

功能函数：
os.remove(path) 删除磁盘中的文件
os.listdir(path) 查看文件夹中的文件，即文件名组成的列表
os.path.exists(path) 判断该路径是否存在，避免出现文件不存在时，程序出现删除错误
"""
path = r'C:\Users\D\Desktop\xml\jpg'
def print_file():
    file = []
    filenames = os.listdir(path)
    for filename in filenames:
        file.append(filename.split(".")[0])
        # print(filename.split(".")[0])
    return file

def print_xml_list():
    j = 0
    xml = []  # xml中存放需要删除的文件名称
    # 示例
    list = ["tz-2021-06-02-14h23m59s011.xml","tz-2021-06-02-14h24m00s086.xml"]
    # 示例
    for i in list:
        if i[-4:] == ".xml":
            xml.append(i[:-4])
            j = j + 1
    return xml


if __name__ == '__main__':
    xml = print_xml_list()  # xml中存放需要删除的文件名称
    print(len(xml))
    file = print_file()  # file中存放所有的文件，包含被需要被删除文件
    print(len(file))
    for i in xml:
        for j in file:
            if i == j and os.path.exists(os.path.join(path + "\\" + i+".jpg")):  # 删除图片：jpg，删除xml：xml
                os.remove(os.path.join(path + "\\" + i+".jpg"))
                print(path + "\\" + i+".xml","被删除")
    print("finish")