如何按首元素对 Python 元组元素分组?


Python 有一个名为 defaultdict 的函数,它按首元素对 Python 元组元素进行分组。

示例

lst = [
   (1, 'Hello', 'World', 112),
   (2, 'Hello', 'People', 42),
   (2, 'Hi', 'World', 200)
]

from collections import defaultdict

d = defaultdict(list)
for k, *v in lst:
   d[k].append(v)
print(d)

输出

这将给出输出

defaultdict(<class 'list'>, {1: [['Hello', 'World', 112]], 2: [['Hello', 'People', 42], ['Hi', 'World', 200]]})

使用 tuple(d.items()) 方法,可以在保持分组的同时将其转换回元组。 

示例

print(tuple(d.items()))

输出

这将给出输出

((1, [['Hello', 'World', 112]]), (2, [['Hello', 'People', 42], ['Hi', 'World', 200]]))

更新时间:2020 年 03 月 05 日

565 次浏览

开启你的 职业生涯

完成课程以获得认证

开始
广告