Python：按長度比較列表的列表並創建大小相等的新列表

Question

我有一個大列表，其中包含多個任意長度的列表。 我想比較每個列表的長度並創建相等大小的新列表。 例如，

biglist = [['x','y','z'],['a','b'],['d','f','g'],...,['r','e','w','q','t','u','i']]
expected_list= [['a','b'],[['x','y','z'],['d','f','g']],....,['r','e','w','q','t','u','i']] 

我是python的新手。 有人可以建議我用一種較便宜的方法來完成上述過程嗎？ 提前致謝。

Answer 1

看起來您想按其元素len s對列表進行groupby ：

>>> biglist = [['x', 'y', 'z'], ['a', 'b'], ['d', 'f', 'g'], ['r', 'e', 'w', 'q', 't', 'u', 'i']]
>>> expected_list = [list(b) for a, b in itertools.groupby(sorted(biglist, key=len), len)]
>>> expected_list
[[['a', 'b']], [['x', 'y', 'z'], ['d', 'f', 'g']], [['r', 'e', 'w', 'q', 't', 'u', 'i']]]

Answer 2

我可以建議使用itertools groupby函數嗎？

import itertools


biglist = [['x','y','z'],['a','b'],['d','f','g'],['r','e','w','q','t','u','i']]

print(list(list(i[1]) for i in itertools.groupby(sorted(biglist, key=len), len)))

哪個輸出

[[['a', 'b']], [['x', 'y', 'z'], ['d', 'f', 'g']], [['r', 'e', 'w', 'q', 't', 'u', 'i']]]