使用Python在同一目录中查找相同类型的文件

Question

Is there a way to find all the files of a particular file type that is specified and recursively call them to be read until you have read all of the files? (This is in Ubuntu)

For example:

In directory /home you have files a.png , b.txt , c.txt , d.txt , e.jpeg and I would like to read all the file that are txt .

How would one do so in Python?

Sub Question: Is it possible to do what was mentioned above with two different file types?

Answer 1

You could use glob.glob() to do this:

>>> from glob import glob

>>> glob('/home/*.txt')
['/home/c.txt', '/home/d.txt', '/home/b.txt']

>>> [glob(i) for i in ['/home/*.txt', '/home/*.jpeg']]
[['/home/c.txt', '/home/d.txt', '/home/b.txt'], ['/home/e.jpeg']]
>>> 

Answer 2

Assuming you're using the extension of the file to guess the type, you could use os.listdir and filter the files based on the filename:

import os

for filename in os.listdir('.'):
    if filename.endswith('.txt'):
        # Do what you want to do