Pandas 按 groupby 上的列标准化

iri*_*ris 6 python normalize pandas pandas-groupby

给定一个熊猫数据框,例如

import pandas as pd

df = pd.DataFrame({'id': ['id1','id1','id2','id2'] , 
                   'x':  [1,2,3,4], 
                   'y':  [10,20,30,40]})
Run Code Online (Sandbox Code Playgroud)

每个数值列可以被归一化到单位间隔[0,1]

columns = ['x', 'y']

for column in columns:
    df[column] = (df[column] - df[column].min()) / (df[column].max() - df[column].min())
Run Code Online (Sandbox Code Playgroud)

导致

    id         x         y
0  id1  0.000000  0.000000
1  id1  0.333333  0.333333
2  id2  0.666667  0.666667
3  id2  1.000000  1.000000
Run Code Online (Sandbox Code Playgroud)

但是,如何在每个数字列上应用这种标准化id?预期的结果将在这个过于简单的例子中

    id         x         y
0  id1  0.000000  0.000000
1  id1  1.000000  1.000000
2  id2  0.000000  0.000000
3  id2  1.000000  1.000000
Run Code Online (Sandbox Code Playgroud)

事实证明不清楚如何在之后更新每个标准化列

df.groupby(['id']).apply(lambda x: ...)
Run Code Online (Sandbox Code Playgroud)

jez*_*ael 4

使用GroupBy.transform

columns = ['x', 'y']
g = df.groupby('id')[columns]
df[columns] = (df[columns] - g.transform('min')) / (g.transform('max') - g.transform('min'))
    
print (df)
    id    x    y
0  id1  0.0  0.0
1  id1  1.0  1.0
2  id2  0.0  0.0
3  id2  1.0  1.0
Run Code Online (Sandbox Code Playgroud)