按组计算百分位数，然后添加到现有数据框中

Question

我有以下数据框

univ  date        ms
A     11/01/2007  0.2
A     11/02/2007  0.3
B     11/01/2007  0.4
B     11/01/2007  0.5

我想要做的是通过 univ 计算百分位数，然后将每个 univ 的值添加到现有数据框中作为额外的列 ms_perc_rank

我可以通过这样做来计算 univ 的百分位数

df.groupby('univ')['ms'].quantile(0.2)

Answer 1

通常，使用transform将groupby结果分配给原始数据帧：

df['ms_perc_rank'] = df.groupby('univ')['ms'].transform('quantile', 0.2)

结果输出：

  univ        date   ms  ms_perc_rank
0    A  11/01/2007  0.2          0.22
1    A  11/02/2007  0.3          0.22
2    B  11/01/2007  0.4          0.42
3    B  11/01/2007  0.5          0.42