如何根据匹配的子串从一个列表创建多个列表？

def create_lists_by_substring(lst, substring):
    lists_dict = {}
    
    for item in lst:
        if substring in item:
            if substring not in lists_dict:
                lists_dict[substring] = []
            lists_dict[substring].append(item)
    
    return list(lists_dict.values())

original_list = ['apple', 'banana', 'orange', 'grape', 'pineapple']
substring = 'a'

result = create_lists_by_substring(original_list, substring)
print(result)

[['apple', 'banana', 'grape'], ['orange']]