python - 如何在Python中对列表进行分组？

Question

我有一个如下所示的列表列表:

['user1', time, cpus, mem]  
['user1', time, cpus, mem]
['user2', time, cpus, mem]
['user3', time, cpus, mem]

等等..

我只需要每个用户一个列表，其中包含时间、CPU 和内存加在一起。我尝试了一些方法，但无法使其发挥作用。

Answer 1

一种甜蜜、Pythonic 且简洁的方法是:

from collections import defaultdict

l = [                  
    ['user1', 0, 1, 0],                                                    
    ['user2', 2, 2, 2],      
    ['user3', 2, 2, 1],
    ['user3', 1, 1, 2],
    ['user1', 1, 0, 1],
]


merged = defaultdict(lambda: [0, 0, 0])

for user, *values in l:               
     merged[user] = [sum(i) for i in zip(values, merged[user])]

输出:

In : merged
Out: 
defaultdict(<function __main__.<lambda>>,
        {'user1': [1, 1, 1], 'user2': [2, 2, 2], 'user3': [3, 3, 3]})

它使用 defaultdict 和长度为 3 的 list 作为默认值。列表中每个元素的相关用户值都会更新。