【Python】Pandas技巧：groupby+agg/transform

公眾號：尤而小屋
作者：Peter
編輯：Peter

本文介紹的是分組groupby分組之后如何使用agg和transform

模擬數(shù)據(jù)

import pandas as pd
import numpy as np

employees = ["小明","小周","小孫","小王","小張"]   # 5位員工
time = ["上半年", "下半年"]


df=pd.DataFrame({
    "employees":np.random.choice(employees,10),  # 在員工中重復(fù)選擇10次
    # 另一種寫法
    #"employees":[employees[x] for x in np.random.randint(0,len(employees),10)],  
    "time":np.random.choice(time,10),
    "salary":np.random.randint(800,1000,10),  # 800-1000之間的薪資選擇10個(gè)數(shù)值
    "score":np.random.randint(6,12,10)  # 6-11的分?jǐn)?shù)選擇10個(gè)
})

df

groupby+單個(gè)字段+單個(gè)聚合

求解每個(gè)人的總薪資金額：

total_salary = df.groupby("employees")["salary"].sum().reset_index()
total_salary

使用agg也能夠?qū)崿F(xiàn)上面的效果：

df.groupby("employees").agg({"salary":"sum"}).reset_index()

df.groupby("employees").agg({"salary":np.sum}).reset_index()

groupby+單個(gè)字段+多個(gè)聚合

求解每個(gè)人的總薪資金額和薪資的平均數(shù)：

方法1：使用groupby+merge

mean_salary = df.groupby("employees")["salary"].mean().reset_index()
mean_salary

然后將上面的兩個(gè)結(jié)果進(jìn)行組合；在合并之前為了字段的名字更加的直觀，我們重命名下：

total_salary.rename(columns={"employees":"total_salary"})
mean_salary.columns = ["employees","mean_salary"]

total_mean = total_salary.merge(mean_salary)
total_mean

方法2：使用groupby+agg

total_mean = df.groupby("employees")\
            .agg(total_salary=("salary", "sum"), 
                 mean_salary=("salary", "mean"))\
            .reset_index()
total_mean

groupby+多個(gè)字段+單個(gè)聚合

針對多個(gè)字段的同時(shí)聚合：

df.groupby(["employees","time"])["salary"].sum().reset_index()

# 使用agg來實(shí)現(xiàn)

df.groupby(["employees","time"]).agg({"salary":"sum"}).reset_index()

groupby+多個(gè)字段+多個(gè)聚合

使用的方法是：

agg(’新列名‘=(’原列名‘, ’統(tǒng)計(jì)函數(shù)/方法‘))

df.groupby(["employees","time"])\
            .agg(total_salary=("salary", "sum"), 
                mean_salary=("salary", "mean"),
                total_score=("score", "sum") 
                )\
            .reset_index()

groupby+transform

關(guān)于transform函數(shù)的使用請參考（建議一定要看）：Pandas高階函數(shù)transform使用機(jī)制

Pandas高級函數(shù)transform使用指南

一張關(guān)于transform函數(shù)的備忘錄：