具有多个条件和操作的 groupby

Question

我有一个问题，我想对数据框进行分组并执行一些操作。输入数据框是这样的：

我想根据 unique_id 进行分组，但如果有两个观察值，我需要一个唯一 ID 中的两行。 1 个 unique_ID 中可能有任意数量的观察结果

预期输出：

我试图计算它的数量和其他列以帮助 groupby，但我没有为我工作。

df["Count_ob"] = df.groupby("Unique_ID")["IF_car_history"].sum()

已获得帮助。谢谢

Answer 1

df = pd.DataFrame({'Unique_id':[1,1,2,2,3,3,3,3],
                   'Car_history':[0,1,0,1,0,1,0,1],
                   'Value':[1000,1500,1000,1200,800,700,1300,1700],
                   'Ob_id':[0,1,0,2,0,3,0,4]})


df["Count_ob"] = df.Ob_id[::-1].cumsum()[::-1] # <=== this do the trick!!!
df["Count_ob"] = df["Count_ob"].max() - df["Count_ob"]

df = df.groupby("Count_ob")[["Unique_id","Car_history","Value","Ob_id"]].agg({'Unique_id':'max',
                                                                        'Car_history':'sum',
                                                                        'Value':'sum',
                                                                        'Ob_id':'sum'}).reset_index(drop=True)
df['Value'] = df.groupby('Unique_id')['Value'].cumsum().values