Question

我遇到了其他一系列加权均值熊猫问题，但似乎没有一个问题可以做我想做的事情。我有以下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 =值*（数量/团队的总数量）

我想我必须以某种方式使用.apply（lambda x：...）函数，但是我不知道如何轻松地获取团队总数量的值。我还遇到了numpy.average函数，但我认为它在这里没有用。

任何帮助将不胜感激！

Answer 1

分为步骤：

>>> 1 + 7
8
>>> (1 + 7) * 3 -.5
23.5
>>> ((1 + 7) * 3 -.5) * ((1 + 7) * 3 -.5)
552.25
>>> import math
>>> math.sqrt(((1 + 7) * 3 -.5) * ((1 + 7) * 3 -.5) + 7)
23.648467180770936

现在，如果您要寻找一种班轮，可以这样做：

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

熊猫的加权平均数据框

1 个答案: