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.

Answer 1

This function returns a list of all file names with the specified extension that live in the specified directory:

import os

def listFiles(path, extension):
    return [f for f in os.listdir(path) if f.endswith(extension)]

print listFiles('/Path/to/directory/with/files', '.txt')

If you want to list all files with the specified extension in a certain directory and its subdirectories you could do:

import os

def filterFiles(path, extension):
    return [file for root, dirs, files in os.walk(path) for file in files if file.endswith(extension)]

print filterFiles('/Path/to/directory/with/files', '.txt')