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

时间:2021-02-07 22:07:53

写这篇博客的缘由:

面试归来翻脉脉发现一个陌生的朋友提出一个面试题,设计实现遍历目录及子目录,抓取.pyc文件。

并贴出两种实现方法:

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

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

个人感觉,这两种方法中规中矩,不像是python的风格。python风格的实现可以考虑python列表推导式。

下面贴出我的实现方法:

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

个人的信条:编程简单粗暴才是真理。