我想按组计算均值,而忽略行本身的值。
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
是该组中所有其他成员的均值,不包括该行自身的值。
答案 0 :(得分:1)
您可以GroupBy
col1
和transform
加上均值。然后从平均值中减去给定行的值:
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的答案进行了修复。