在pandas groupby中对列表中的行进行分组
发布于2019-08-21 19:06 阅读(717) 评论(0) 点赞(5) 收藏(4)
我有一个像熊猫一样的数据框：
a b
A 1
A 2
B 5
B 5
B 4
C 6我希望按第一列分组，并将第二列作为行中的列表：
A [1,2]
B [5,5,4]
C [6]使用pandas groupby可以做这样的事吗？
解决方案
您可以使用groupby以对感兴趣的列进行分组，然后apply list对每个组进行分组：
In [1]:
# create the dataframe
df = pd.DataFrame( {'a':['A','A','B','B','B','C'], 'b':[1,2,5,5,4,6]})
df
Out[1]:
a b
0 A 1
1 A 2
2 B 5
3 B 5
4 B 4
5 C 6
[6 rows x 2 columns]
In [76]:
df.groupby('a')['b'].apply(list)
Out[76]:
a
A [1, 2]
B [5, 5, 4]
C [6]
Name: b, dtype: object所属网站分类: 技术文章 > 问答
作者：黑洞官方问答小能手
链接：https://www.pythonheidong.com/blog/article/51017/462b427944dae4e63661/
来源：python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容：(最多支持255个字符)
---无人问津也好，技不如人也罢，你都要试着安静下来，去做自己该做的事，而不是让内心的烦躁、焦虑，坏掉你本来就不多的热情和定力