我需要根据分组结果添加新列。以以下数据框为例。
ID City
234x Lima
342x Rica
234x Rio
333x NYC
333x SF
我将使用group_by来获取计数:
df_GroupBy = pd.DataFrame({'count':df.groupby([ "ID"]).size()}).reset_index()
这会产生一个输出:
ID Count
234x 2
342x 1
333x 2
我现在想要做的是获取这个输出:
ID City City_2
234x Lima Rio
342x Rica
333x NYC SF
我已经尝试过使用Transform
和map
,但没有太大的成功。感谢你的帮助。