首页 > 解决方案 > groupby 并提取数据

问题描述

pandas 中的新功能,我能够从 csv 文件创建数据框。我也能够整理出来。

我现在正在苦苦挣扎的是:我以 pandas 数据框中的图像为例。 一个数据框

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

我想根据第二列在同一个唯一数据框中取出第三列。

我强调几个例子:对于数字 9,返回序列 [60,61,70,51]

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

对于数字 8,取回单个元素 8。

如何使用 groupby 进行此提取?

非常感谢亚历克斯

标签: pythonpandassortingpandas-groupby

解决方案


从这个问题的答案开始,我们可以提取以下代码来获得所需的结果。

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)

推荐阅读