使用 Python 计算值组的标准偏差

Question

我的数据与此类似：

index name number difference
0     AAA  10     0
1     AAA  20     10
2     BBB  1      0
3     BBB  2      1
4     CCC  5      0
5     CCC  10     5
6     CCC  10.5   0.5

我需要计算基于名称组的差异列的标准偏差。

我试过了

data[['difference']].groupby(['name']).agg(['mean', 'std'])

和

data["std"]=(data['difference'].groupby('name').std())

但两者都为传递给groupby（）的变量提供了KeyError。我试图用以下方法解决它：

data.columns = data.columns.str.strip()

但错误仍然存在。

提前谢谢。

Answer 1

您可以先对整个数据框使用 groupby（['name']），并且只对感兴趣的列应用 agg：

data = pd.DataFrame({'name':['AAA','AAA','BBB','BBB','CCC','CCC','CCC'],
                    'number':[10,20,1,2,5,10,10.5],
                    'difference':[0,10,0,1,0,5,0.5]})
data.groupby(['name'])['difference'].agg(['mean', 'std'])