python - pandas 如何對上千萬資料分組「快速」取第一行和最後一行？

Question

我希望取pandas分組後每個分組的第一行和最後一行，目前的做法是遍歷分組for name,group in xxx:然後group.iloc[0] group.iloc[-1]，但這對上千萬的資料遍歷起來很慢，能不能有什麼快速的方法？

Answer 1

可以使用DataFrame的groupbys方法傳回物件的first和last方法來進行取得