pandas的排序、分組groupby及cumsum累計求和方式

更新時間：2022年05月17日 11:12:04 作者：種麥南山下

這篇文章主要介紹了pandas的排序、分組groupby及cumsum累計求和方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

生成一列sum_age 對age 進行累加

df['sum_age'] = df['age'].cumsum()
print(df)

生成一列sum_age_new 按照 gender和is_good 對age進行累加

df['sum_age_new'] = df.groupby(['gender','is_good'])['age'].cumsum()
print(df)

根據(jù)不同的性別對年齡進行等級排序

df['rank_g'] = df.groupby(['gender'])['age'].rank()
print(df)

這里的 rank( ) 即 'rank_g' ，并不是按照1、2、3、4、、依次排

按照官方文檔的意思，該函數(shù)是沿著某個軸來計算數(shù)值數(shù)據(jù)等級(1到n)。默認情況下，為相等的值分配同一個等級，該等級是這些值的等級的平均值。

例子：

import pandas as pd
obj = pd.Series([7,-5,7,4,2,0,4])
print(obj.rank())

代碼對 [7, -5, 7, 4, 2, 0, 4] 進行從小到大地排序，很明顯地，可以排成 [-5, 0, 2 ,4, 4, 7, 7]，數(shù)值7有第6和第7兩個位置，那應該排序應該排到第幾級？根據(jù)官方文檔，取平均值，(6+7)/2=6.5，所以兩個7的等級都為6.5，同理可得兩個4的等級都為(4+5)/2=4.5。

輸出：

0 6.5
1 1.0
2 6.5
3 4.5
4 3.0
5 2.0
6 4.5
dtype: float64

對數(shù)據(jù)排序之后，分組，并累計求和

# 對Start Time進行排序，Connection Type分組，temp進行累計求和cumsum
wsw_1 = wsw.sort_values(['Start Time'])
wsw_1.loc[:, 'Connection Number'] = wsw_1.groupby(['Connection Type'])['temp'].cumsum()

這里如果不對start time排序，Connection Number不會按時間順序，統(tǒng)計drilling、tripping 的number數(shù)

pandas分組排序功能

在一個班級里，學生考試科目有語文、數(shù)學、英語，分別有對應的成績。

現(xiàn)在，想要列出每個科目班級的前五名的情況，要求包含科目、姓名、成績、名次。

通過以下代碼實現(xiàn)：

import pandas as pd
a=['小紅','小綠','小藍','小白','小青','小紫','小粉','小傻','小紅','小綠','小藍','小白','小青','小紫','小粉','小傻','小紅','小綠','小藍','小白','小青','小紫','小粉','小傻']
b=['語文','語文','語文','語文','語文','語文','語文','語文','數(shù)學','數(shù)學','數(shù)學','數(shù)學','數(shù)學','數(shù)學','數(shù)學','數(shù)學','英語','英語','英語','英語','英語','英語','英語','英語']
c=[97,65,23,43,67,23,55,98,56,45,67,78,98,45,87,65,67,23,55,98,56,45,67,78]
len(a),len(b),len(c)
df=pd.DataFrame({'name':a,'kemu':b,'score':c})
df2=df.sort_values(['kemu','score','name'], ascending=[1, 0,1])
df2['rn']=df2.groupby(['kemu']).rank(method='first',ascending =0)['score']
df2[df2['rn']<=5]
''''

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: