Python:如何查找具有特定扩展名的所有文件？

Question

我试图.c使用Python 查找目录中的所有文件.

我写了这个,但它只是将所有文件归还给我 - 而不仅仅是.c文件.

import os
import re

results = []

for folder in gamefolders:
    for f in os.listdir(folder):
        if re.search('.c', f):
            results += [f]

print results

我怎样才能获得.c文件？

Answer 1

尝试将内循环更改为这样的东西

results += [each for each in os.listdir(folder) if each.endswith('.c')]

Answer 2

试试"glob":

>>> import glob
>>> glob.glob('./[0-9].*')
['./1.gif', './2.txt']
>>> glob.glob('*.gif')
['1.gif', 'card.gif']
>>> glob.glob('?.gif')
['1.gif']

Answer 3

有一个更好的解决方案，直接使用正则表达式，它是标准库fnmatch处理文件名模式的模块。（另请参阅glob模块。）

编写一个辅助函数：

import fnmatch
import os

def listdir(dirname, pattern="*"):
    return fnmatch.filter(os.listdir(dirname), pattern)

并按如下方式使用它：

result = listdir("./sources", "*.c")

Answer 4

for _,_,filenames in os.walk(folder):
    for file in filenames:
        fileExt=os.path.splitext(file)[-1]
        if fileExt == '.c':
            results.append(file)

Answer 5

吻

# KISS

import os

results = []

for folder in gamefolders:
    for f in os.listdir(folder):
        if f.endswith('.c'):
            results.append(f)

print results