Фильтровать файлы по имени файла

Question

У меня есть папка, которая имеет 4 разных файла. Например:

Тип 1: 00001_a.png

Тип 2: 00231_b.mat

Тип 3: 00001_c.jpg

Тип 4: 00001_c.png

Как я могу фильтровать эти файлы в 4 списках? Мое текущее решение может фильтровать только на основе расширения файла.

all_file = os.walk(input_path).next()[2] #get files only 
list_one = [ fi for fi in all_file if fi.endswith("*.png") ] # "*_a.png" won't work 

Answer 1

Просто опустить звездочки (*) в endswith() и он будет работать, как ожидалось, например, fi.endswith('_a.png').

Предлагаемое лучшее решение, которое позволяет избежать жесткого кодирования поддерживаемых типов:

from collections import defaultdict 

def get_file_type(filename): 
    base, ext = os.path.splitext(filename) 
    return base.rsplit('_', 1)[1] + ext 

files_by_type = defaultdict(list) 
for filename in os.listdir(input_path): 
    filetype = get_file_type(filename) 
    files_by_type[filetype].append(filename) 

Answer 2

Рассмотрим регулярное выражение решения с использованием списка каталога os модуля:

import os, re 

# CURRENT DIRECTORY OF RUNNING SCRIPT (OR MANUALLY ENTER PATH) 
cd = os.path.dirname(os.path.abspath(__file__)) 

a_pngfiles = [file for file in os.listdir(cd) if re.match("^.*_a.png$", file)] 
b_matfiles = [file for file in os.listdir(cd) if re.match("^.*_b.mat$", file)] 
c_jpgfiles = [file for file in os.listdir(cd) if re.match("^.*_c.jpg$", file)] 
c_pngfiles = [file for file in os.listdir(cd) if re.match("^.*_c.png$", file)]