python - 如何计算具有分类值的组内百分比？

Question

我有一个数据框:

id   value_type
1       b
1       a
1       a
2       a
3       a
3       b

我想计算每个 value_type 与每个 id 组的百分比。所以期望的结果是:

id   value_type       perc
1       b             0.33
1       a             0.66
2       a             1
3       a             0.5
3       b             0.5

我该怎么做？我尝试了 groupby().size() 但它很重要，但我需要百分比

Answer 1

检查下面的代码:

import pandas as pd

df = pd.DataFrame({'col1':[1,1,1,2,3,3],'col2':['b','a','a','a','a','b']})

df['perc'] = df.groupby(['col1','col2'])['col2'].transform('count')/df.groupby('col1')['col2'].transform('count')

df.round(2).drop_duplicates()

输出: