对不同组的 python pandas 进行计算

Question

我有一个包含不同组的熊猫数据集。对于每个组，我想应用不同的计算。最好的方法是什么？例如，输入数据Group val1 val21      12   31      19   42      23   93      59   743      2    44输出数据Group val1 val2 output1      12   3   3*val1*val21      19   4   3*val1*val22      23   9   5*val1*val23      59   74  10*val1*val23      2    44  10*val1*val2

Answer 1

您需要创建一个映射/字典d将组链接到其计算值。接下来，map它到列Group并使用 numpy ufunc `reduce of multiply 创建所需的输出

import numpy as np

d = {1: 3, 2: 5, 3: 10}

df['output'] = np.multiply.reduce([df.Group.map(d), df.val1, df.val2])

Out[243]:

   Group  val1  val2  output

0      1    12     3     108

1      1    19     4     228

2      2    23     9    1035

3      3    59    74   43660

4      3     2    44     880

如果你不想使用numpy，只需对每一列进行乘法

df['output'] = df.Group.map(d) * df.val1 * df.val2

Answer 2

如果该字符串是您正在寻找的文字输出，那么您可以为每个组创建一个字典并映射值。然后只需将字符串添加到末尾：

dct = {1:3, 2:5, 3:10}

df['output'] = df['Group'].map(dct).astype(str) + '*val1*val2'

df

Out[1]: 

   Group  val1  val2        output

0      1    12     3   3*val1*val2

1      1    19     4   3*val1*val2

2      2    23     9   5*val1*val2

3      3    59    74  10*val1*val2

4      3     2    44  10*val1*val2

现在，我逐字记录了你的输出，但如果你试图将这些值相乘:)，那么你可以这样实现：

dct = {1:3, 2:5, 3:10}

df['output'] = df['Group'].map(dct) * df['val1'] * df['val2']

df

Out[1]: 

   Group  val1  val2  output

0      1    12     3     108

1      1    19     4     228

2      2    23     9    1035

3      3    59    74   43660

Answer 3

我想你可以使用 apply

df.groupby([“group”, “val1”, “val2”]).apply(lambda x: x)