Python Pandas组中的最大值为新列

Question

我正在尝试计算一个新列,其中包含多个组中每个组的最大值.我来自Stata背景,所以我知道Stata代码会是这样的:

by group, sort: egen max = max(odds) 

例如:

data = {'group' : ['A', 'A', 'B','B'],
    'odds' : [85, 75, 60, 65]}

然后我希望它看起来像:

    group    odds    max
     A        85      85
     A        75      85
     B        60      65
     B        65      65

最后,我想形成列需要1/(max-min) * odds地方max和min是为每个组.

Answer 1

使用groupby+ transform：

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

这相当于冗长：

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

该transform方法将groupby结果与groupby索引器对齐，因此不需要显式映射。

Answer 2

df['max'] = df.group_col.map(lambda x: df.groupby('group_col').odds.max()[x])