在python，求下列要求的速度快的方法：有一个pandas的df，需要将其分组，然后在每个分组中求最大值与平均值的比值，再将这个比值添加到df中

Question

比如，有一个df，是ABCD四个班共10个小孩的年龄：
import pandas as pd
import numpy as np

data_dict = { "class":['A','B','A','C','D','B','C','A','D','C'] , 'age':[ 8, 11, 9, 12, 16, 6, 7, 10, 13, 5]}

data_df = pd.DataFrame(data_dict)

现在，需要将data_df按班级分组，求得每个班级的最大的年龄与每个班级的平均年龄的比值"ratio_age"，并且data_df新增一列，以添加“ratio_age”。请问怎么做，才能够达到运算时速度最快（尽可能最快）？

Answer 1

import pandas as pd

data_dict = { "class":['A','B','A','C','D','B','C','A','D','C'] , 'age':[ 8, 11, 9, 12, 16, 6, 7, 10, 13, 5]}
data_df = pd.DataFrame(data_dict)

data_df['ratio_age'] = data_df.groupby('class')['age'].transform(lambda x: max(x) / (sum(x) / len(x)))
print(data_df)