使用pandas dataFrame中的Groupby日期创建列表列表

Question

df =

        Date     Slot
0   2022-02-23    34
1   2022-02-23    35
2   2022-02-24     0
3   2022-02-24     1
4   2022-02-25     0
5   2022-02-25     1

这是我的df，我想要一个具有相同相应的“日期”值的所有“插槽”值的列表，即我的输出应该看起来像[[34,35]，[0,1]，[0,1]，[0，[0， 1]]

ps：日期列中的值未经事先知道，因此在这里没有进行硬编码，是否有一般性执行此任务的概括。

Answer 1

您可以尝试

out = df.groupby('Date')['Slot'].agg(list).tolist()
#
out = df.groupby('Date')['Slot'].apply(list).tolist()

print(out)

[[34, 35], [0, 1], [0, 1]]