fnmatch模块

fnmatch() 函数匹配能力介于简单的字符串方法和强大的正则表达式之间。 如果在数据处理操作中只需要简单的通配符就能完成的时候，这通常是一个比较合理的方案

import os
from fnmatch import fnmatchcase

files = os.listdir('.')
py_files = [file for file in files if fnmatchcase(file, '*.py')]
print(py_files)