Question

I am trying to find all the .c files in a directory using Python.

I wrote this, but it is just returning me all files – not just .c files:

import os
import re

results = []

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

print results

How can I just get the .c files?

Answer 1

try changing the inner loop to something like this

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