python - pandas 如何对上千万数据分组'快速”取第一行和最后一行？

Question

我希望取pandas分组后每个分组的第一行和最后一行，目前的做法是遍历分组for name,group in xxx:然后 group.iloc[0] group.iloc[-1]，但这对上千万的数据遍历起来很慢，能不能有什么快速的方法？

Answer 1

可以使用DataFrame的groupbys方法返回对象的first和last方法来进行获取