微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何在groupby函数的pandas系列中找到比率

我使用groupby按职业和性别对数据集进行分组.现在我想找到每个职业的男女比例.我无法想到如何继续.

enter image description here

解决方法:

这是使用pandas.pivot_table和矢量化Pandas计算的一种方法.请注意,此方法无需执行单独的groupby.

df = pd.DataFrame([['A', 'F'], ['A', 'F'], ['A', 'M'], ['B', 'M'], ['B', 'M'], ['B', 'F'],
                   ['C', 'M'], ['C', 'M'], ['D', 'F']], columns=['Occupation', 'Gender'])

# pivot input dataframe
res = df.pivot_table(index='Occupation', columns='Gender', aggfunc='size', fill_value=0)

# calculate ratios
sums = res[['F', 'M']].sum(axis=1)
res['FemaleRatio'] = res['F'] / sums
res['MaleRatio'] = res['M'] / sums

print(res)

Gender      F  M  FemaleRatio  MaleRatio
Occupation                              
A           2  1     0.666667   0.333333
B           1  2     0.333333   0.666667
C           0  2     0.000000   1.000000
D           1  0     1.000000   0.000000

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。

相关推荐