在熊猫数据框中使用分组日期创建列表列表

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] ]

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]]