熊猫:计算均值而忽略自己行的值

时间:2019-04-16 13:41:56

标签: python pandas mean aggregation

我想按组计算均值,而忽略行本身的值。

import pandas as pd

d = {'col1': ["a", "a", "b", "a", "b", "a"], 'col2': [0, 4, 3, -5, 3, 4]}
df = pd.DataFrame(data=d)

我知道如何按组返回均值:

df.groupby('col1').agg({'col2': 'mean'})

哪个返回:

Out[247]: 
  col1  col2
1    a     4
3    a    -5
5    a     4

但是我想要的是按组的意思,而忽略了行的值。例如。第一行:

df.query('col1 == "a"')[1:4].mean()

返回:

Out[251]: 
col2    1.0
dtype: float64

编辑: 预期的输出是与上述df格式相同的数据帧,其中的列mean_excl_own是该组中所有其他成员的均值,不包括该行自身的值。

2 个答案:

答案 0 :(得分:1)

您可以GroupBy col1transform加上均值。然后从平均值中减去给定行的值:

df['col2'] = df.groupby('col1').col2.transform('mean').sub(df.col2)

答案 1 :(得分:0)

感谢您的所有输入。我最终使用了@VnC链接的方法。

这是我解决的方法:

import pandas as pd

d = {'col1': ["a", "a", "b", "a", "b", "a"], 'col2': [0, 4, 3, -5, 3, 4]}
df = pd.DataFrame(data=d)

group_summary = df.groupby('col1', as_index=False)['col2'].agg(['mean', 'count'])
df = pd.merge(df, group_summary, on = 'col1')

df['other_sum'] = df['col2'] * df['mean'] - df['col2'] 
df['result'] = df['other_sum'] / (df['count']  - 1)

查看最终结果:

df['result']

哪些印刷品:

Out: 
0    1.000000
1   -0.333333
2    2.666667
3   -0.333333
4    3.000000
5    3.000000
Name: result, dtype: float64

编辑:以前我在列名方面遇到了一些麻烦,但是我使用this的答案进行了修复。