聚合 Function 到 dataframe，同時保留 Pandas 中的行

Question

我想根據一個名為COLLISION_ID的字段和每個COLLISION_ID的計數來聚合我的數據。

我想刪除重復的 COLLISION_ID，因為它們具有相同的坐標，但保留原始數據集中的出現次數。

我的代碼如下

df2 = df1.groupby(['COLLISION_ID'])[['COLLISION_ID']].count()

這將返回：

我希望我的數據以COLLISION_ID數字、計數和我的數據的剩余列的形式返回，這些列未在此處顯示（大約 40 列稍后將被過濾）

Answer 1

如果您在談論過濾器，我們應該進行transform

df1['count_col']=df1.groupby(['COLLISION_ID'])['COLLISION_ID'].transform('count')

然后您可以使用列數過濾 df1