按列对数据框进行分组并在另一个中连接字符串

Question

我知道这应该很容易，但它让我发疯......

我正在尝试将数据框转换为分组数据框。

df 输出：

    Postcode    Borough             Neighbourhood
0   M3A         North York          Parkwoods
1   M4A         North York          Victoria Village
2   M5A         Downtown Toronto    Harbourfront
3   M5A         Downtown Toronto    Regent Park
4   M6A         North York          Lawrence Heights
5   M6A         North York          Lawrence Manor
6   M7A         Queen's Park        Not assigned
7   M9A         Etobicoke           Islington Avenue
8   M1B         Scarborough         Rouge
9   M1B         Scarborough         Malvern
10  M3B         North York          Don Mills North
...

我想创建一个分组数据框，其中 Neighborhood 按 Postcode 分组，然后所有邻域变成 Neighborhoods 的串联字符串，按 Postcode 分组... 类似：

    Postcode    Borough             Neighbourhood
0   M3A         North York          Parkwoods
1   M4A         North York          Victoria Village
2   M5A         Downtown Toronto    Harbourfront, Regent Park
...

我正在尝试使用：

df.groupby(['Postcode'])['Neighbourhood'].apply(lambda strs: ', '.join(strs))

但这不会返回新的数据帧..当我在运行后使用df时，它会输出相同的原始数据帧。

如果我使用：

df = df.groupby(['Postcode'])['Neighbourhood'].apply(lambda strs: ', '.join(strs))

它把 df 变成一个对象？

Answer 1

使用此代码

new_df = df.groupby(['Postcode', 'Borough']).agg({'Neighbourhood':lambda x:', '.join(x)}).reset_index()

reset_index() 将按列从索引中取出您的分组，并将其作为列返回到数据框并创建一个新的整数索引。