如何将数据框中的行转换为 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]