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

Question

我正在尝试使用 Python 在目录中查找所有 .c 文件。

我写了这个，但它只是将所有文件返回给我 - 而不仅仅是 .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')]