python - 打印 groupby 对象的内容

Question

我有一张像这样的 table

fruit       day
---------------------
apple       Mon
apple       Mon
apple       Tue
banana      Tue

我想根据水果进行分组，并在列表中包含唯一的日期，并在另一列中列出列表的长度。我熟悉如何进行 groupby，但不确定如何获取列表中的这些唯一值并获取其长度。寻找类似的结果

fruit     day_list    count
apple     [Mon, Tue]  2 
banana    [Tue]       1

Answer 1

检查groupby和unique + nunique

df.groupby('fruit').day.agg(['unique','nunique']).reset_index()
Out[242]: 
    fruit      unique  nunique
0   apple  [Mon, Tue]        2
1  banana        [Tu]        1