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

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 groupbyfrom 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 meanl = [(('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