pandas_VS_Excel提取各班前2名后2名的数据
【要求】
- 提取各班前2名的数据
- 提取各班后2名的数据
【代码】
代码语言：javascript复制# -*- coding: utf-8 -*-
'''
提取出了分组中的前2名：例如：提取出各班的总分的前2名
提取出分组的中的后2名：例如：提取出各班的总分的后2名
'''
import pandas as pd
df=pd.read_excel('数据源（5个班各6人）.xlsx')
#这里先插入一个列'班名次'方便自己提取出数据后进行观察
df['班名次']=df['总分'].groupby(df['班别']).rank(ascending=False)
print(df.sort_values(by=['班别','总分'],ascending=[True,False]))
#print(df)
df_q2=df.sort_values('总分',ascending=False).groupby('班别').head(2)
print(df_q2)
df_h2=df.sort_values('总分',ascending=False).groupby('班别').tail(2)
print(df_h2)【解析】
- 先用分组再rank()插入一列，标记出班名次，方便观察
- 取前2名：先用总分排名，再用groupby分组，取各分组的前2个数据
- 取后2名：先用总分排名，再用groupby分组，取各分组的后2个数据
【效果】
- 标记 “班名次”
- 取前2名
- 取后2名
- 若有需要，可以输出到excel文件中的 ====今天就学习到此====
