python - 带有熊猫的加权平均数据框
问题描述
我遇到了一堆其他加权平均熊猫问题,但他们似乎都没有做我想做的事情。我有以下df:
Primary_Key Team Quantity Value 1 Value 2
0 A Blue 10 20 10
1 B Red 5 19 30
2 C Green 8 13 29
3 D Blue 12 24 18
4 E Red 15 25 19
5 F Green 12 18 23
我正在尝试计算每个团队的每个值的加权平均值,所以我会得到以下 result_df:
Team Quantity Value 1 Value 2
0 Blue 10 20*10/22 10*10/22
1 Red 5 19*5/20 30*5/20
2 Green 8 13*8/20 29*8/20
3 Blue 12 24*12/22 18*12/22
4 Red 15 25*15/20 19*15/20
5 Green 12 18*12/20 23*12/20
其中“值”列下的每个条目都对其进行了以下计算:
weighted_mean = value * (数量/团队总数量)
我想我必须以某种方式使用 .apply(lambda x:...) 函数,但我不知道如何轻松获得团队总数量的值。我也遇到了 numpy.average 函数,但我认为它在这里没有用。
任何帮助将非常感激!
解决方案
Breaking down into steps:
import pandas as pd
import numpy as np
df = pd.DataFrame(data={
'Primary_Key': list('ABCDEF'),
'Team': ['Blue', 'Red', 'Green', 'Blue', 'Red', 'Green'],
'Quantity': [10,5,8,12,15,12],
'v1': [20,19,13, 24,25,18],
'v2': [10,30,29,18,19,23]})
df['GroupQuantity'] = df.groupby('Team')['Quantity'].transform(np.sum)
df['v1'] = df['Quantity'] * df['v1'] / df['GroupQuantity']
df['v2'] = df['Quantity'] * df['v2'] / df['GroupQuantity']
df
Primary_Key Team Quantity v1 v2
0 A Blue 10 9.090909 4.545455
1 B Red 5 4.750000 7.500000
2 C Green 8 5.200000 11.600000
3 D Blue 12 13.090909 9.818182
4 E Red 15 18.750000 14.250000
5 F Green 12 10.800000 13.800000
Now if you are looking for a one liner, you can do:
df[['v1', 'v2']] = df[['v1', 'v2']] * df['Quantity'].to_numpy()[:,None] / df.groupby('Team')['Quantity'].transform(np.sum).to_numpy()[:,None]
df
Primary_Key Team Quantity v1 v2
0 A Blue 10 9.090909 4.545455
1 B Red 5 4.750000 7.500000
2 C Green 8 5.200000 11.600000
3 D Blue 12 13.090909 9.818182
4 E Red 15 18.750000 14.250000
5 F Green 12 10.800000 13.800000
推荐阅读
- java - 为什么在运行时垂直调整 Frame 大小后 JRadioButtons 会出现故障?
- swift - 创建一个快速帮助类来处理 CoreLocation 函数
- sql - 多表信息的条件比较
- swift - 我是在创建两个单例对象还是只是一个参考?
- kubernetes - K8s http Liveness 探测仅在 1 个工作节点上失败
- php - Laravel 基于数组从数据库中选择相关数据
- python - 如何从一个 .txt 文件中提取一堆行并将它们添加到另一个文件中?
- javascript - 我可以使用哪个正则表达式在标签内查找电子邮件地址?
- elixir - 如何在我的本地机器上运行 mozilla 集线器?
- pdf - 有没有办法使用 MS Graph API 对我们的 WOPI 主机提供的文档进行操作?