python查找目錄及子目錄下特定文件

寫這篇博客的原因:python

面試歸來翻脈脈發現一個陌生的朋友提出一個面試題,設計實現遍歷目錄及子目錄,抓取.pyc文件。面試

並貼出兩種實現方法:編程

我的感受,這兩種方法中規中矩,不像是python的風格。python風格的實現能夠考慮python列表推導式。spa

下面貼出個人實現方法:設計

1 def getFiles(path, suffix):
2     return [os.path.join(root, file) for root, dirs, files in os.walk(path) for file in files if file.endswith(suffix)]

我的的信條:編程簡單粗暴纔是真理。code

相關文章
相關標籤/搜索