Python: How can I find all files with a particular extension?

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')] 

Answer 2

Try "glob":

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