python—查找以XXX結尾的文件

# 查找電腦所有視頻
for cur_dir,dirs,files in os.walk(r'f:'):
    print('當前正在%s目錄下查找'%cur_dir)
    for f in files:#當前目錄下的所有文件
        if f.endswith('.mp4'):
            real_path = os.path.join(cur_dir,f)
            #os.remove(real_path)
            print('發現一個小電影【%s】，目錄正在 【%s】'%(f,cur_dir))