python - Pandas ，按功能分组后的列名称

Question

我有一个名为 purchase_cat_df 的简单 Pandas Dataframe:

              email                cat
0  email1@gmail.com  Mobiles & Tablets
1  email2@gmail.com  Mobiles & Tablets
2  email1@gmail.com  Mobiles & Tablets
3  email3@gmail.com  Mobiles & Tablets
4  email3@gmail.com      Home & Living
5  email1@gmail.com      Home & Living

我按“电子邮件”分组并将“猫”放入如下列表中:

test = purchase_cat_df.groupby('email').apply(lambda x: list(x.cat))

但是我的 DataFrame 测试是:

email
email1@gmail.com    [Mobiles & Tablets, Mobiles & Tablets, Home & ...
email2@gmail.com                                  [Mobiles & Tablets]
email3@gmail.com                   [Mobiles & Tablets, Home & Living]

我丢失了索引和名称，如何命名第 2 列？

Answer 1

如果您想保留原始索引，您可能正在寻找这样的东西:

purchase_cat_df.groupby('email', as_index=False)

as_index=False 保留原始索引。然后您可以继续按名称对该列进行寻址。