python - 根据 Python 中的键获取排序列表的一部分

Question

是否可以根据某个键值(例如列表项的长度)对排序列表进行切片？如果是，怎么办？

例如，我得到一个这样的排序列表:

sorted_list = sorted(some_list, key=len)

现在我想得到一个切片，其中包含具有最低和相等 len 的所有项目(即 len 的所有项目:min(sorted_list, key=len) )，它应该位于排序列表。

Answer 1

您可以先对项目进行分组，然后获取第一个结果子迭代器的元素。

from itertools import groupby


firsts = list(next(groupby(sorted(some_list, key=len), len))[1])

例如，

>>> some_list = [[1, 2, 3], [4, 5, 6], [1], [2], [2, 3]]
>>> list(next(groupby(sorted(some_list, key=len), len))[1])
[[1], [2]]