pandas习题 182：每组最大绩效的员工完整记录

看过来

《pandas 教程》持续更新中，提供建议、纠错、催更等加作者微信: gr99123（备注：pandas教程）和关注公众号「盖若」ID: gairuo。跟作者学习，请进入 Python学习课程。欢迎关注作者出版的书籍：《深入浅出Pandas》和《Python之光》。

（编码题）按部门分组，返回每组最大绩效的员工完整记录。人事部 8 月 6 条员工绩效：

import pandas as pd

df = pd.DataFrame({
    "员工": ["张三", "李四", "王五", "赵六", "孙七", "周八"],
    "部门": ["技术", "技术", "市场", "市场", "技术", "市场"],
    "绩效": [85, 90, 78, 82, 88, 80]
})

用 一行 groupby().apply() 找出 各部门绩效最高的员工完整记录，并打印结果。

参考答案

代码如下：

top = df.groupby("部门").apply(lambda g: g.loc[g["绩效"].idxmax()])
print(top)

输出示例：

      员工  部门  绩效
部门                 
技术   李四  技术   90
市场   赵六  市场   82

（完）

pandas习题 182：每组最大绩效的员工完整记录

参考答案

相关内容