多种分组聚合

import pandas as pd
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
#这两个参数的默认设置都是False  用于将数据对齐展示
pd.set_option('display.unicode.ambiguous_as_wide', True)
pd.set_option('display.unicode.east_asian_width', True)

data = pd.read_csv('../dataset/5/5.4/data.csv')
data.head()

 

 

#单字段分组
print(data.groupby('省份')['月消费（元）'].agg('sum'))
print('============================================')
print(data.groupby('省份')['月消费（元）'].sum())
print('==============================================')
print(data.groupby('省份').agg({'月消费（元）':'sum'}))
#前面三个效果相同

print('===========================================')
print(data.groupby('省份')['月消费（元）'].agg({'费用':'sum'}))

#这是为了定义一个列名，最后使用reset_index()将多重索引变成一个

#多字段分组，单字段聚合
print(data.groupby(['省份','手机品牌'])['月消费（元）'].agg({'费用':'sum'}))  #reset_index()

 

#多字段分组，多字段聚合
print(data.groupby(['省份','手机品牌'])['月消费（元）','月流量（M）'].agg('sum').reset_index().head(10))
print('=============================')

#多字段分组，多字段聚合，多个聚合函数
print(data.groupby(['省份','手机品牌'])['月消费（元）','月流量（M）'].agg({'总计':'sum','平均':'mean'}).head(10))
print('=========================')
print(data.groupby(['省份','手机品牌'])['月消费（元）','月流量（M）'].agg(['sum','mean']).head(10))

#多个字段各自使用自己的函数

gp = data.groupby(['省份','手机品牌']).agg({
    '月消费（元）':{'求和':'sum','平均值':'mean','最小':'min'},
    '月流量（M）':{'求和':'sum','平均值':'mean','最小':'min'}
}).head(10)
gp

 

 

# index0 =gp.columns.get_level_values(0)
# index1 =gp.columns.get_level_values(1)
# print(index0)
# print(index1)
# gp.columns = index0 + '_' +index1
gp.reset_index()