Python代碼，提取擴展

Question

可能重複：
In python, how can I check if a filename ends in '.html' or '_files'?

import os 

path = '/Users/Marjan/Documents/Nothing/Costco' 

print path 

names = os.listdir(path) 

print len(names) 

for name in names: 
    print name 

這是我一直在使用的代碼，它列出了這一類終端中的所有名稱。這個文件中有幾個文件名（Costco）沒有.html和_files。我需要挑選出來，唯一的問題是它有超過2500個文件名。需要關於代碼的幫助，該代碼將搜索此路徑並挑選出所有不以.html或_files結尾的文件名。謝謝你們

Answer 1

for name in names: 
    if filename.endswith('.html') or filename.endswith('_files'): 
     continue 
    #do stuff 

通常，如果你需要一個文件的擴展名os.path.splitext()會更合適，但在這種情況下endswith()是完全沒有問題。

Answer 2

比ThiefMaster的建議短一些：

for name in [x for x in names if not x.endswith(('.html', '_files'))]: 
    # do stuff