python - 从一维列表创建列表列表，按长度对每个元素进行分组

Question

假设我有一个简单的列表:

flat_list = ["one", "a", "two", "b", "ab"]

我的目标是获取一个列表列表，每个列表包含具有相同长度的每个元素，如下所示:

list_lists = [["one", "two"], ["a", "b"], ["ab"]]

是否可以通过简单的列表理解来完成此任务？

Answer 1

使用itertools.groupby为了实现这一点(首先我们使用 sorted 按长度对列表进行排序，以便 groupby 按照我们想要的方式工作):

[list(group) for key, group in groupby(sorted(flat_list, key=len, reverse=True), len)]

<小时/>

>>> from itertools import groupby
>>> flat_list = ["one", "two", "a", "b", "ab"]
>>> list_of_lists = [list(group) for key, group in groupby(sorted(flat_list, key=len, reverse=True), len)]
>>> 
>>> list_of_lists
[['one', 'two'], ['ab'], ['a', 'b']]
>>>