Python Pandas max value in a group as a new column

后端 未结 3 1669
温柔的废话
温柔的废话 2020-12-07 00:51

I am trying to calculate a new column which contains maximum values for each of several groups. I\'m coming from a Stata background so I know the Stata code would be somethi

3条回答
  •  天命终不由人
    2020-12-07 01:28

    Use groupby + transform:

    df['max'] = df.groupby('group')['odds'].transform('max')
    

    This is equivalent to the verbose:

    maxima = df.groupby('group')['odds'].max()
    df['max'] = df['group'].map(maxima)
    

    The transform method aligns the groupby result to the groupby indexer, so no explicit mapping is required.

提交回复
热议问题