首页 > 解决方案 > 在 Python 中操作数据框

问题描述

假设我在 Python 中有以下数据框

可乐 colB
1
2
bb 3
bb 4

我怎样才能得到以下

可乐 colB
{1,2}
bb {3,4}

标签: python

解决方案


您可以使用以下聚合将数据框分组为一组:

df_agg = df.groupby('colA').agg(set).reset_index()

推荐阅读