pandas-分组

概述

本文介绍groupby() 分组函数

学过sql语句的同学,可以理解为sql中的group by 分组语句

语法

data_frame.groupby(by='')
参数
  • by:指定列名,表示按照该列进行分组。
    多列分组时,list类型
返回
  • DataFrameGroupBy类型的对象

例子

泰坦尼克数据,对性别进行分组

import pandas as pd
df=pd.read_csv("train.csv")

dfgb=df.groupby(by='Sex')
print(dfgb)

结果如下:

<pandas.core.groupby.groupby.DataFrameGroupBy object at 0x0000000B980B5E48>

返回对象gbo是DataFrameGroupBy数据类型

DataFrameGroupBy类型

遍历

import pandas as pd

df=pd.read_csv("train.csv")
dfgb=df.groupby(by='Sex')
for i,j in dfgb:
    print(i,type(i))
    print('-'*50)
    print(j,type(j))
    print('='*50)

从结果可知,i是性别的值,j是该性别的记录,是DataFrame类型

对多列进行分组

import pandas as pd

df=pd.read_csv("train.csv")
dfgb=df.groupby(by=['Sell_No','Prices'])

原文出处:/show_1EF3EzrcDUXq.html