python groupby用法实战

py3study

发布于 2020-01-07 10:32:46

1.2K0

发布于 2020-01-07 10:32:46

文章被收录于专栏：python3

背景

已知某航空公司拥有的飞机数量如下图示，现要求在EXCEL表中，计算出各个机型每种座位布局的飞机数量在该种机型飞机总量中所占的比重。

代码实现

#导入原始数据表
import pandas as pd
dt=pd.read_excel('xl.xlsx')
#定义函数per，即子数占总数的比
def per(arr):
    return arr/arr.sum()
#利用GROUPBY对机型进行分组，再利用per()计算各组内数据占该组数据之和的比重，并把所得结果添加到dt数据框的后一列，保存为lx.xlsx
dt[u'占比']=dt.groupby(u'机型').transform(per)
dt.to_excel('lx.xlsx',index=False)

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2019/09/17 ，如有侵权请联系 cloudcommunity@tencent.com 删除

el