python - 避免数据帧 groupby 结果变成系列

Question

嗨，我在数据帧上做了 groupby 和 nlargest。结果是一个系列，无论如何将结果保留为数据框？

ID NO
0 10
0 20
0 30
1 20
1 30
1 40
2 30
3 40

我的代码 TOP2 = df.groupby('ID')['NO'].nlargest(2)。

它返回正确的结果，但我希望它是一个数据框，而不是系列。

标签： pythondataframe