Pandas: apply: использование лямбды для создания списка кортежей

У меня есть следующий фрейм данных 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 для каждого идентификатора. Код работает нормально.

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

Однако этот вывод дает мне только первый (stage, days) каждого идентификатора, и это не похоже на кортеж:

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

Я что-то здесь пропустил? Спасибо!