python 根据python中指定的正则表达式对字符串列表进行分组

Posted 2021-05-09

from re import search, findall

def group_strings(string_iterable, pattern_to_group_with):
    """ groups strings based on a given regex and returns the groups in a dict
        by: Cody Kochmann """
    groups={}
    for i in string_iterable:
        if search(pattern_to_group_with, i):
            group = findall(pattern_to_group_with, i)[0]
            if group in groups:
                groups[group].append(i)
            else:
                groups[group]=[i]
    return groups