python——读取某一目录及其子目录下所有的同类型文件

import os

def gather_files(root_dir, keep_suffixs=['jpg', 'png']):
    """
    根据文件后缀递归获取root_dir下所有的该类型文件
    输出：数组，每个元素是一个文件路径
    """
    file_counts = {}
    for suffix in keep_suffixs:
        file_counts[suffix] = 0
    n_files = 0
    all_files = []
    for parent, dirnames, filenames in os.walk(root_dir, followlinks=False):
        for filename in filenames:
            for suffix in keep_suffixs:
                if filename.endswith(suffix):
                    file_counts[suffix] += 1
                    file_path = os.path.join(parent, filename)
                    all_files.append(file_path)
                    n_files += 1
                    break
    return all_files