Скажем, у меня есть пандас DataFrame, такой как:

   A  B  id
0  1  1   0
1  2  1   0
2  3  2   1
3  0  2   1

Скажем, я хочу объединить строки с одним и тем же идентификатором, чтобы другие элементы в строках были объединены в список, чтобы приведенный выше кадр данных стал:

     A       B     id
0  [1, 2]  [1, 1]   0
1  [3, 0]  [2, 2]   1

как первые две строки, и последние две строки имеют одинаковый идентификатор. Есть ли у панды функция для этого? Мне известна команда pandas groupby, но я бы хотел, чтобы возвращаемый тип также был в виде фрейма данных. Благодарю.