按给定元素对列表进行分组并计算平均值

Question

如果有以下列表：l = [(('01001', '01003'), 4.15),  (('01001', '01005'), 2.83),  (('01001', '01007'), 3.32),  (('01002', '01009'), 6.83),  (('01002', '01011'), 2.53),  (('01003', '01013'), 20.50),  (('01003', '01013'), 20.50)]我想计算此列表的第一个元素（“01001”、“01002”和“01003”）的平均值。到目前为止，我的方法是将列表转换为数据框，并将第一个括号分成两列，然后计算平均值。有没有更Pythonic的方式？任何帮助将非常感激！

Answer 1

熊猫对此似乎有些过分了。由于要分组的这些值似乎是连续的（否则使用字典代替），因此您可以使用itertools.groupby并取这些浮点数的平均值：

from itertools import groupby

from statistics import mean

[mean(v for *_, v in v) for k,v in groupby(l, lambda x: x[0][0])]

# [3.4333333333333336, 4.68, 20.5]

或者，如果您更喜欢字典：

{k:mean(v for *_, v in v) for k,v in groupby(l, lambda x: x[0][0])}

# {'01001': 3.4333333333333336, '01002': 4.68, '01003': 20.5}

Answer 2

像这样的东西：

from statistics import mean

l = [(('01001', '01003'), 4.15),

 (('01001', '01005'), 2.83),

 (('01001', '01007'), 3.32),

 (('01002', '01009'), 6.83),

 (('01002', '01011'), 2.53),

 (('01003', '01013'), 20.50),

 (('01003', '01013'), 20.50)]

 

print(mean(int(x[0][0]) for x in l))

输出

1001.8571428571429