人生苦短,自学 python——pandas 的分组操作


四类基本操作之分组

索引、分组、变形、合并

例数据说明,作为资深科密,我将使用kaggle上老大职业生涯的投篮数据为例,理论结合案例说明分组运算的基本原理和一些基本操作。老大职业生涯数据统计,其中有5000条数据为空,是当时比赛用来预测的结果数据,因此,该数据与老科真实数据之间会有一定的差异。

此外,数据中没有给出每场的得分,但是可以根据shot_type和shot_made_flag两个字段得到得分数据,具体数据处理过程不是本文重点,这里就不再详述了,数据说明:

一、分组运算的原理


1、SAC过程

SAC指的是分组操作中的split-apply-combine过程其中,split指基于某一些规则,将数据拆成若干组,apply是指对每一组独立地使用函数,combine指将每一组的结果组合成某一类数据结构


2、 apply过程

整合(Aggregation)——分组计算统计量(如求均值、求每组元素个数)

变换(Transformation)——分组对每个单元的数据进行操作(如元素标准化)

过滤(Filtration)——按照某些规则筛选出一些组(如选出组内某一指标小于50的组)

综合问题——前面提及的三种问题的混合


二、groupby函数


1、分组原理及结果

可根据一列、多列、多级索引的level以及与数据长度相同的数组列表等进行等值分组,经过groupby后会生成一个groupby对象,该对象本身不会返回任何数据,只有当相应的方法被调用才会起作用.

比如将老科的比赛数据按照常规赛和季后赛进行分组

可以看出数据被分成了两组0:常规赛,1:季后赛,将原来一个整体的dataframe分为两个dataframe


2、组对象的常用方法调用

get_group()获得组结果,

size()和ngroups():组容量与组数,

head()和first():对分组对象使用head函数,返回的是每个组的前几行,而不是数据集前几行;first显示的是以分组为索引的每组的第一个分组信息


3、分组依据

分组的依据是非常自由的,只要是与数据框长度相同的列表即可,同时支持函数型分组。根据奇偶行分组:


df.groupby(lambda x:'奇数行' if not df.index.get_loc(x)%2==1 else '偶数行').groups

从原理上说,我们可以看到利用函数时,传入的对象就是索引


4、groupby的[]操作

可以用[]选出groupby对象的某个或者某几个列,比如对比科比常规赛和季后赛投篮得分比。


kobe.groupby(["playoffs","game_id"])["points"].sum().groupby(level=0).mean()
playoffs
0    15.76791
1    16.43578
Name: points, dtype: float64

这里因为数据缺失得到的结果与老科的实际数据有偏差。


三、聚合、过滤和变换


1、聚合(Aggregate)

(1)常用聚合函数

所谓聚合就是把一堆数,变成一个标量,因此
mean/sum/size/count/std/var/sem/describe/first/last/nth/min/max都是聚合函数

计算科比生涯对阵对手得分情况

(2)多聚合函数

利用元组进行重命名

(3)多列应用聚合函数

单列多聚合函数,结果列名为聚合函数名,多列应用聚合函数,结果列名为原列名,可以使用rename函数进行重命名。

(4)自定义聚合函数

(5)带参数的聚合函数


2、过滤(Filteration)

filter函数是用来筛选某些组的(务必记住结果是组的全体)全组值均为true的组,因此传入的值应当是布尔标量


3、变换(Transformation)

transform函数中传入的对象是组内的列,并且返回值需要与列长完全一致,如果返回了标量值,那么组内的所有元素会被广播为这个值。


四、apply函数


当函数不是聚合、过滤、变换,而是更普通的函数时,就要用到apply函数,返回值可以为标量、列表,数据框等形式

1、函数应用

(1)标量返回值

此时的功能可视为agg聚合函数

(2)列表返回值

(3)数据框返回值

2、用apply同时统计多个指标


相关文章

Python算法:4.寻找两个正序数组的中位数

题目:寻找两个正序数组的中位数给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。 算法的时间复杂度应该为 O(log(m+n...

在Python中怎么用二分法插入一个新的数到数组列表

接昨天的把一个数插入到有序的数组、列表中,我们今天使用二分法来,就好比一个队分成2个队,因为有序的么,先比较中间的,得到结果,只要比较一个方向的一半就可以了,这样一来是不是减少一半的时间,节约是运行时...

Python实现【分割数组的最大差值】

n = int(input()) nums = list(map(int, input().split())) total_sum = sum(nums) max_diff = 0 left_sum...

Python实现【找出两个整数数组中同时出现的整数】

from collections import defaultdict import sys def solve(): # 读取输入 arr1 = list(map(int, sys...

Python实现分治算法?

分治算法(Divide and Conquer Algorithm)是一种设计算法的策略,它将一个问题分成多个相似的子问题,递归地解决这些子问题,然后将结果合并以得到原问题的解。典型的分治算法包括归并...

【找出两个整数数组中同时出现的整数】Python实现

from collections import Counter def find_common_elements(arr1, arr2): # 统计数组中每个数字的出现次数 coun...