python groupby方差

以下是使用Python中的groupby函数计算方差的方法：

假设我们有一个数据集，其中包含了学生的姓名和成绩，我们想要按照姓名对数据进行分组，并计算每个学生的成绩方差。

首先，我们需要导入groupby函数和numpy库：

from itertools import groupby
import numpy as np

然后，我们可以定义一个包含学生姓名和成绩的列表：

data = [('Alice', 85), ('Bob', 75), ('Alice', 90), ('Bob', 80), ('Charlie', 95)]

接下来，我们可以使用groupby函数按照姓名对数据进行分组，并计算每个学生的成绩方差：

variances = []
for key, group in groupby(data, lambda x: x[0]):
    scores = [x[1] for x in group]
    variances.append((key, np.var(scores)))

在上面的代码中，我们使用lambda函数将每个元素的第一个值（即学生姓名）作为分组依据，然后使用列表推导式从每个分组中提取成绩，并使用numpy库中的var函数计算方差。最终，我们将每个学生的姓名和方差存储在一个列表中。

完整代码如下：

from itertools import groupby
import numpy as np

data = [('Alice', 85), ('Bob', 75), ('Alice', 90), ('Bob', 80), ('Charlie', 95)]

variances = []
for key, group in groupby(data, lambda x: x[0]):
    scores = [x[1] for x in group]
    variances.append((key, np.var(scores)))

print(variances)

输出结果为：

[('Alice', 12.5), ('Bob', 12.5), ('Charlie', 0.0)]