pandas組內(nèi)排序,并在每個(gè)分組內(nèi)按序打上序號(hào)的操作
問題:
pandas組內(nèi)排序,并在每個(gè)分組內(nèi)按序打上序號(hào)
描述:
pandas dataframe 對(duì)dep_id組內(nèi)的salary排序。希望給下面原本只有前三列的dataframe,添加上第四列。
等價(jià)于sql里的排序函數(shù) row_number() over() 功能

假設(shè)我已經(jīng)建好了僅有前三列的dataframe,數(shù)據(jù)集命名為 MyData,
解決方案如下:
MyData['sort_id'] = MyData['salary'].groupby(MyData['dep_id']).rank()
結(jié)果如下:

補(bǔ)充:Pandas.DataFrame實(shí)現(xiàn)分組、排序并且為分組插入排名
1. 示例數(shù)據(jù)(各班級(jí)學(xué)生得分)
import pandas as pd
data_dict = {"name":
["Rose", "Jack", "Tom", "Kyle", "Jery", "Adam", "Bill", "David", "Denny", "Evan"],
"class": [1, 2, 3, 1, 2, 3, 1, 2, 3, 1],
"score": [88, 92, 38, 98, 22, 65, 45, 53, 97, 100]}
df = pd.DataFrame(data=data_dict)
df

2. 按班級(jí)分組
df = df.groupby('class', sort=False)\
.apply(lambda x:x.sort_values("score", ascending=False))\
.reset_index(drop=True)
df

3. 給各分組班級(jí)增加排名列
df["rank"] = None
# 標(biāo)識(shí)班級(jí)
flag = df.loc[0].values[1]
rank = 0
for i in range(len(df)):
temp = df.loc[i].values[1]
if (temp == flag).all():
# 同一班級(jí)
rank += 1
else:
# 不同班級(jí),重新計(jì)算排名
flag = temp
rank = 1
df.loc[i, "rank"] = rank
df

以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教。
相關(guān)文章
Django 拼接兩個(gè)queryset 或是兩個(gè)不可以相加的對(duì)象實(shí)例
這篇文章主要介紹了Django 拼接兩個(gè)queryset 或是兩個(gè)不可以相加的對(duì)象實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-03-03
python統(tǒng)計(jì)字符串中指定字符出現(xiàn)次數(shù)的方法
這篇文章主要介紹了python統(tǒng)計(jì)字符串中指定字符出現(xiàn)次數(shù)的方法,涉及Python中count函數(shù)的使用技巧,非常具有實(shí)用價(jià)值,需要的朋友可以參考下2015-04-04
Python OpenCV機(jī)器學(xué)習(xí)之圖像識(shí)別詳解
OpenCV中也提供了一些機(jī)器學(xué)習(xí)的方法,例如DNN等。本文將為大家詳細(xì)介紹一下OpenCV中利用機(jī)器學(xué)習(xí)實(shí)現(xiàn)的一些圖片識(shí)別功能:人臉識(shí)別、車牌識(shí)別等,感興趣的可以了解一下2022-01-01
Python數(shù)據(jù)處理利器Pandas?DataFrame常用操作
這篇文章主要為大家介紹了Python數(shù)據(jù)處理利器Pandas?DataFrame,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-06-06
Python3實(shí)現(xiàn)的簡單工資管理系統(tǒng)示例
這篇文章主要介紹了Python3實(shí)現(xiàn)的簡單工資管理系統(tǒng),涉及Python文件讀寫、數(shù)據(jù)遍歷、判斷等相關(guān)操作技巧,需要的朋友可以參考下2019-03-03
python 實(shí)現(xiàn)批量xls文件轉(zhuǎn)csv文件的方法
今天小編就為大家分享一篇python 實(shí)現(xiàn)批量xls文件轉(zhuǎn)csv文件的方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2018-10-10

