|
本帖最后由 sheeboard 于 2021-2-1 19:00 编辑
参考
- import pandas as pd
- df1=pd.read_excel('测试数据.xlsx')
- df2=pd.read_excel('每个学校取的人数.xlsx')
- result=pd.DataFrame()
- for ind,data in df2.iterrows():
- school_name=data['学校']
- num=data['人数']
- temp=df1[df1['学校']==school_name]
- temp.sort_values('总分',ascending=False,inplace=True)
- result.append(temp.head(num)ignore_index=True)
- result.to_excel('result.xlsx',index=False)
复制代码 |
|