删除文件夹中的txt或者其他特定类型文件

import os
path = r''
fileFinalName = ''
def delTxt(path):
    lists = os.listdir(path)
    for file in lists :
         if file.split('.')[-1] == fileFinalName:
             os.remove(path+'\\'+file)


if __name__ == '__main__':
    delTxt(path)