Python:将相同的整数值分组，然后求平均值

Question

我有大量数据，我想对相同整数的值进行分组，然后取平均值。

例如:

a = [0, 0.5, 1, 1.5, 2, 2.5]

我想按如下方式分组:

[0, 0.5] [1, 1.5] [2, 2.5]

...然后取平均值并将所有平均值放入一个新数组中。

Answer 1

假设您想按数字的整数值进行分组(因此数字向下舍入)，类似这样的操作可能会起作用:

>>> a = [0, 0.5, 1, 1.5, 2, 2.5]
>>> groups = [list(g) for _, g in itertools.groupby(a, int)]
>>> groups
[[0, 0.5], [1, 1.5], [2, 2.5]]

然后平均变为:

>>> [sum(grp) / len(grp) for grp in groups]
[0.25, 1.25, 2.25]

这假设 a 已经排序，如您的示例所示。

引用号:itertools.groupby , list comprehensions .