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已经排序，如您的示例所示。

Answer 2

如果您使用其他库没有问题：

import pandas as pd

import numpy as np

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

print(pd.Series(a).groupby(np.array(a, dtype=np.int32)).mean())

给出：

0    0.25

1    1.25

2    2.25

dtype: float64

Answer 3

如果您想要使用字典的方法，您可以像这样继续：

dic={}

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

for items in a:

    if int(items) not in dic:

        dic[int(items)]=[]

    dic[int(items)].append(items)

print(dic)

for items in dic:

    dic[items]=sum(dic[items])/len(dic[items])

print(dic)

Answer 4

您可以使用groupby轻松获得它（您可能需要先对列表进行排序）：

from itertools import groupby

from statistics import mean

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

for k, group in groupby(a, key=int):

    print(mean(group))

会给：

0.25

1.25

2.25