Python | 列表的分组展平

在Python中，有时候我们需要将一个嵌套的列表进行展开，使其变成一个扁平的列表，而展开的同时需要将相同的元素进行分组。下面我们来分享一下如何使用Python的语法实现这个任务。

方法一：使用itertools库中的chain()函数

在Python的itertools库中有一个非常好用的函数叫做chain()，可以将多个可迭代对象合并在一起，组成一个单一的迭代器，所以我们可以先将所有的子列表都合并在一起，然后再对其进行去重和分组操作。

import itertools

def flat_group(lst):
    flat_lst = list(itertools.chain.from_iterable(lst))
    return {item: flat_lst.count(item) for item in sorted(set(flat_lst))}

lst = [[1, 2], [3, 4, 1], [2, 3]]
result = flat_group(lst)
print(result)  # {1: 2, 2: 2, 3: 2, 4: 1}

方法二：使用collections库中的Counter()函数

Python的collections库中有一个Counter()函数，可以方便地统计文本中各单词出现的次数。我们可以将每个子列表使用join()函数连接成一个字符串，再将所有的字符串合并在一起，最后再使用Counter()函数进行计数和分组操作。

from collections import Counter

def flat_group(lst):
    flat_lst = Counter(' '.join(map(str, lst)).split())
    return dict(flat_lst)

lst = [[1, 2], [3, 4, 1], [2, 3]]
result = flat_group(lst)
print(result)  # {1: 2, 2: 2, 3: 2, 4: 1}

总结

以上就是Python对嵌套列表进行展开并对相同元素进行分组的方法，使用Python的内置函数或者库函数可以非常简单地实现这个任务。希望本文对大家学习Python有所帮助。