分组并提取数据

Question

是熊猫的新功能，我可以从csv文件创建数据框。我也能够解决它。

我现在正在努力的事情如下：我以一个熊猫数据框的图像为例。

第一列是索引， 第二列是组号 第三栏是发生了什么。

我希望基于第二列在相同的唯一数据帧上取出第三列。

我将举几个例子：对于数字9，返回序列 [60,61,70,51]

对于数字6取回序列 [65,55,56]

对于数字8，返回单个元素8。

如何使用groupby进行提取？

非常感谢 问候 亚历克斯

Answer 1

从有关question的答案开始，我们可以提取以下代码以接收所需的结果。

dataframe =  pd.DataFrame({'index':[0,1,2,3,4], 'groupNumber':[9,9,9,9,9], 'value':[12,13,14,15,16]})
grouped = dataframe.groupby('groupNumber')['value'].apply(list)