python如何遍历某种特定类型的文件

netdzb

#!/usr/bin/env python3
# -*- coding:utf-8 -*-

import os

def getFiles(dir, suffix): # 查找根目录，文件后缀
    res = []
    for root, directory, files in os.walk(dir):  # =>当前根,根下目录,目录下的文件
            for filename in files:
                    name, suf = os.path.splitext(filename) # =>文件名,文件后缀
                    if suf == suffix:
                            res.append(os.path.join(root, filename)) # =>吧一串字符串组合成路径
    return res

for file in getFiles("./", '.py'):  # =>查找以.py结尾的文件
            print(file)

======================

上面的代码把当前的所有给列出来了，我想列出扩展名是001,002,003,004......的文件该怎么做？

for file in getFiles("./", '.0*'):  # =>查找以.py结尾的文件
改成上面的没有用的。

小渣飞

下面的for循环再加个判断呀