Как преобразовать строки в кадре данных в несколько списков в Python

Question

У меня большой кадр данных следующего типа:

Part    Text
P1  a
P1  b
P2  c
P2  d

Я хочу преобразовать эти данные в два формата

Ожидаемый вывод 1:

Part    Text
P1  ['a','b']
P2  ['c','d']

Ожидаемыйвывод 2:

Part    Text
P1  ['ab']
P2  ['cd']

Может кто-нибудь помочь мне с этим

Answer 1

Использование groupby.agg:

df1 = df.groupby('Part')['Text'].agg(list).reset_index()
df2 = df.groupby('Part')['Text'].agg(lambda x: [''.join(x.values)]).reset_index()

print(df1)
  Part    Text
0   P1  [a, b]
1   P2  [c, d]

print(df2)
  Part  Text
0   P1  [ab]
1   P2  [cd]