将列中的值转换为单行Python

Question

我有一个看起来像这样的熊猫数据框：

Area1  Area2 
  1      2      
  1      4
  1      5
  1      9
  2      8
  2      16
  2      4
  2      1
  3      8
  3      9

如何转换“ Area2”列，使其成为每个“ Area1”列的值列表

所以我想要的输出是：

Area1     Area2 
  1      2, 4, 5, 9     
  2      8, 16, 4, 1
  3      8, 9

我以前在R中完成过此操作：

df %>% group_by(Area1) %>% summarise(Area2= toString(sort(unique(Area2)))) 

我一直在尝试groupby（）和agg（），但是没有成功。

当我使用df.groupby（'Area1'）对数据进行分组后，有人可以解释一下我可以使用什么

非常感谢您的任何建议。

Answer 1

您可以分组并应用列表

import pandas as pd
df=pd.read_csv("test.csv")
df.groupby('Area1')['Area2'].apply(list)

Answer 2

R代码段执行字符串连接。

以下行保留了Area2的原始类型。

import pandas as pd

df.groupby('Area1').Area2.apply(pd.Series.tolist).reset_index()