pandas_VS_Excel提取各班前2名后2名的数据

# -*- 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)