python - Pandas 中多索引的百分比

Question

我需要找到 MultiIndex 列(“计数”)的百分比。我的数据框看起来像:

            count
A  week1     264

   week2      29

B  week1     152

   week2      15

我想添加一个“百分比”列来制作

            count percent
A  week1     264      0.9

   week2      29      0.1

B  week1     152     0.91

   week2      15     0.09

我知道我可以找到我想要的总数

mydf.sum(level=[0, 1])

但我似乎无法弄清楚如何将其转换为百分比。

Answer 1

您可以使用 groupby 和 transform 来做到这一点:

df['percent'] = df.groupby(level=0).transform(lambda x: (x / x.sum()).round(2))

#          count  percent
# A week1    264     0.90
#   week2     29     0.10
# B week1    152     0.91
#   week2     15     0.09