如何使用AGG获得第一位和第二个最大值？

Question

我有一个数据框架：

id   type1           val
a    main             3
a    main             5
a    main             4
b    second           80
b    second           90

我想在Val的列ID type1和Max（）上进行组，并获得最大观察的相应索引。我这样做：

df.groupby(["id","type1"])["val"].agg("max", "idxmax")

我得到了：

id   type1        max   idxmax   

a    main          5    1
b   second        90    4

但是我想获得第一位和第二个最大值：

id   type1        max   idxmax   

a    main          5    1
a    main          4    2
b   second        90    4
b    second       80    3

该怎么做？

Answer 1

使用nlargest（n）（在这里文档）

df.groupby(["id","type1"])["val"].agg("nlargest", 2)

id  type1    
a   main    1     5
            2     4
b   second  4    90
            3    80