Yes; use the .apply(...)
function, which will be called on each sub-DataFrame
. For example:
grouped = df.groupby(keys)
def wavg(group):
d = group['data']
w = group['weights']
return (d * w).sum() / w.sum()
grouped.apply(wavg)
Yes; use the .apply(...)
function, which will be called on each sub-DataFrame
. For example:
grouped = df.groupby(keys)
def wavg(group):
d = group['data']
w = group['weights']
return (d * w).sum() / w.sum()
grouped.apply(wavg)