Pandas aggregate with function with multiple parameters

Question

df = pd.DataFrame({'a': list('aabb'), 'b': [1, 2, 3, 4], 'c': [4, 2, 1, 3]})

def agg_func(x, y):
    return sum(x) > sum(y)

I need smth like that

df.groupby('a').agg({('b', 'c'): agg_func})

And expect

mozway · Accepted Answer · 2022-07-20 09:51:54Z

2

Here I would simple aggregate the columns, and then perform the comparison:

g = df.groupby('a').sum()
out = g['b']>g['c']

With your function (probably less efficient):

out = df.groupby('a').apply(lambda g: agg_func(g['b'], g['c']))

output:

a    False
b     True
dtype: bool

answered Jul 20, 2022 at 9:51

mozway

267k13 gold badges56 silver badges106 bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1