在数组python中创建数组

Question

我有一个这样的数据帧:

<表类> ID 日期 等级 信息—— tbody><<tr>1212/51安妮1212/52路易1212/53尼科1212/54维克1315/52斯蒂芬妮1315/53雅典娜1315/54路易

使用.agg:

res = df.groupby(["ID", "Date"])["Rank"].agg(list).tolist()
print(res)

[[1, 2, 3, 4], [2, 3, 4]]

如果list的numpy.array是预期输出，则可选:

res = df.groupby(["ID", "Date"])["Rank"].agg(list).to_numpy()
print(res)

[list([1, 2, 3, 4]) list([2, 3, 4])]