以下是一个示例代码,演示了如何按照列表中的元素进行分组。
from collections import defaultdict
def group_by_elements(lst):
groups = defaultdict(list)
for element in lst:
groups[element].append(element)
return list(groups.values())
# 示例用法
lst = [1, 2, 3, 1, 2, 3, 4, 5]
result = group_by_elements(lst)
print(result)
输出:
[[1, 1], [2, 2], [3, 3], [4], [5]]
在上面的示例中,我们使用defaultdict
来创建一个默认值为列表的字典对象groups
。然后,我们遍历列表lst
中的每个元素,并将元素添加到对应的分组中。最后,我们使用list()
函数将字典的值转换为一个列表,并返回结果。
在示例中,输入的列表lst
包含了重复的元素,因此输出的分组结果中,重复的元素被放在了同一个分组中。