Python: Как группировать файлы префикс

Question

У меня есть набор файлов:

a_a_a_morestuff 
a_a_a_morestuff 
a_a_b_morestuff 
a_a_b_morestuff 
a_a_c_morestuff 
a_b_b_morestuff 
a_b_b_morestuff 

Как я могу использовать питон перебрать их и манипулировать кластеры файлов, которые имеют один и тот же префикс т.е., a_a_a? Я не буду заранее знать, сколько файлов у меня есть или сколько у меня будет одного и того же префикса.

Answer 1

Использование glob

import glob 
print glob.glob('my_directory/a_a_a*') 

выход ниже, где my_directory содержит много файлов, в том числе a_a_a_X где X является 1-4.

['my_directory/a_a_a_3', 'my_directory/a_a_a_2', 'my_directory/a_a_a_1', 'my_directory/a_a_a_4'] 

Edit: Если вы не ориентируетесь на конкретный префикс, но знать формат префикса, вы можете использовать этот код:

import os 
import re 

files = os.listdir("somedirectory") 

prefixes = {} 

for f in files: 
    m = re.search('[^_]+_[^_]+_[^_]+_', f) 
    if m: 
     prefix = m.group(0) 
     print prefix 
     if prefix in prefixes: 
      prefixes[prefix].append(f) 
     else: 
      prefixes[prefix] = [f] 

for prefix in prefixes: 
    print prefixes[prefix] 

Для случая, когда файлы является следующий массив :

files = [ 
    'out-15.03_.03_12.57_001.h5', 
    'out-15.03_.03_25.05_001.h5', 
    'out-15.03_.03_25.05_002.h5', 
    'out-15.03_.03_25.05_003.h5', 
    'out-15.03_.03_37.52_001.h5' ] 

Это выход:

['out-15.03_.03_12.57_001.h5'] 
['out-15.03_.03_25.05_001.h5', 'out-15.03_.03_25.05_002.h5', 'out-15.03_.03_25.05_003.h5'] 
['out-15.03_.03_37.52_001.h5']