Pandas数据帧行按日期合并为一行

Question

以以下pandas数据帧为例：

date,test
2022-09-01,demo1
2022-09-01,demo2
2022-09-01,demo3
2022-09-02,demo4
2022-09-02,demo5
2022-09-02,demo6

我想转换到这个数据帧，其中"日期"的每个值都是唯一的，并且插入了相同日期的所有列：

date,test, test, testx, testy ...
2022-09-01,demo1, demo2, demo3 ...
2022-09-02,demo4, demo5, demo6 ...

有什么想法吗？，非常感谢

Answer 1

尝试：

df["tmp"] = df.groupby("date").cumcount() + 1
out = df.pivot(index="date", columns="tmp", values="test").add_prefix("test_")
out.columns.name, out.index.name = None, None
print(out)

打印：

test_1 test_2 test_3
2022-09-01  demo1  demo2  demo3
2022-09-02  demo4  demo5  demo6