python—分组取top

最新推荐文章于 2023-06-06 21:05:35 发布

selinamay

最新推荐文章于 2023-06-06 21:05:35 发布

阅读量1.6k

点赞数

文章标签： python 开发语言后端

本文链接：https://blog.youkuaiyun.com/selinamay/article/details/122325890

版权

这篇博客介绍了如何利用Pandas的apply函数处理数据分组时取前三名的问题，特别是当存在并列第三名的情况。作者通过定义top3函数，实现了保留并列第三名的所有记录。这个函数首先获取每组的前三个元素，然后检查剩余数据中是否存在与第三名相同值的记录，将这些记录一并加入最终结果。最后，应用这个函数到数据集的每个分组上，得到满足条件的聚合结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

写代码时遇上一个分组取top的问题，要求分组取前三，但如果后面有跟第三并列的数值也要纳入进来，因此考虑使用apply函数。

def top3(df):
    df_1 = df.head(3)
    df_2 = df_1.iloc[-1] #前n条的最后一条
    df_3 = pd.concat([df, df_1]).drop_duplicates(keep=False) #去除前n条的数据
    df_4 = df_3[(df_3['到期收益率(%)']==df_2['到期收益率(%)'])] #在未选入的数据中查找
    df_fin = pd.concat([df_1,df_4])
    return df_fin

final3 = final2.groupby('发行人全称').apply(top3)

通过apply对分组的各组进行操作。