python - 如何计算熊猫的调和平均值

Question

我有一个如下所示的数据框。Words指每封电子邮件发送的字数。

sender receiver words 
a        b       10
a        c       5
a        c       15
b        a       50
b        a       30

我对熊猫比较陌生。我想计算 1）每对之间发送的电子邮件数量 2）两个人之间发送的总字数的调和平均值。如何使用hmean()fromscipy.stats来获得所需的输出？

sender  receiver  total_emails  total_words
   a        b                   hmean([10])
   a        c                   hmean([5,15])
   b        a                   hmean([50,30])

对于电子邮件的总数，我不确定正确的公式应该是什么。任何帮助，将不胜感激！

标签： pythonpandasscipy

Answer 1

你可以使用groupby：

from scipy import stats
df = df.groupby(['sender', 'receiver']).agg(stats.hmean).reset_index(name='total_words')

输出：

  sender receiver  total_words
0      a        b         10.0
1      a        c          7.5
2      b        a         37.5