Преобразование Groupby в список в пандах не работает ⇐ Python

[Anonymous]

Лучше всего описать на примере

Код: Выделить всё

import pandas as pd

df = pd.DataFrame({
'a'   : ['A','B','C','A','B','C','A','B','C'],
'b': [1,2,3,4,5,6,7,8,9]}
)

И я хочу создать столбец, содержащий в списке элементы столбца b по группе столбца a

что приведет к следующему

Код: Выделить всё

   a    b      c
0  A    1  [1, 4, 7]
1  A    4  [1, 4, 7]
2  A    7  [1, 4, 7]
3  B    2  [2, 5, 8]
4  B    5  [2, 5, 8]
5  B    8  [2, 5, 8]
6  C    3  [3, 6, 9]
7  C    6  [3, 6, 9]
8  C    9  [3, 6, 9]

Я могу сделать это с помощью groupby и применить или agg, а затем объединить фреймы данных вот так

Код: Выделить всё

df_tmp = df.groupby('a')['b'].agg(list).reset_index()
df.merge(df_tmp, on='a')

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

Код: Выделить всё

df['c'] = df.groupby('a')['b'].transform(list)

но столбец c такой же, как столбец b

А также следующее

Код: Выделить всё

df.groupby('a')['b'].transform(lambda x: len(x))

возвращает серию со значениями 3 т.е. длина сгруппированных элементов равна 3 (чего и следовало ожидать)

А еще это

Код: Выделить всё

df.groupby('a')['b'].transform(lambda x: list(x))

не дает ожидаемого результата.

Итак, к моему вопросу, как я могу получить желаемый результат с помощью groupby и tranform

Код: Выделить всё

pandas

версия 1.0.5

Подробнее здесь: https://stackoverflow.com/questions/624 ... s-not-work