熊猫:应用:使用lambda创建元组列表

我有以下数据框 df :

    id  stage days
--------------------
    a1   A    1
    a2   A    3
    a3   A    2
    a4   A    5
    a1   B    1
    a2   B    2
    a1   C    2
    a3   D    3

我在 Pandas GroupBy 对象我通过聚合 stage 每个id的列。代码工作正常。

df1 = df.groupby('id').apply(lambda x: list(x['stage'])).reset_index() 
df1

输出如下所示:

a1  [A, B, C]
a2  [A, B]
a3  [A, D]
a4  [A]

现在我想为每个 id 组问题是 (stage, days) 。我修改了上述代码,如下所示:

df2 = df.groupby('id').apply(lambda x:list((x['stage'],x['days']))).reset_index() 
df2

我想要 df2 就像:

a1  [(A, 1), (B, 1), (C, 2)]
a2  [(A, 3), (B, 2)]
a3  [(A, 2), (D, 3)]
a4  [(A, 5)]

然而,这个输出只给了我第一个 (阶段,天) 它看起来不像一个元组:

a1  [[A], [1]]
a2  [[A], [3]]
a3  [[A], [2]]
a4  [[A], [5]]

我错过什么了吗?谢谢