pandas - 如何制作基于 2 列的嵌套列表？

Question

我有一个数据框。我想制作一个基于另一列的列的嵌套列表。

col1      col2
As     K
SD     k
JK     K
LO     P
WQ     p
IU     A

嵌套列表的每个元素都是 col1 中的项目列表在 col2 中具有相同的值.
输出

Nlist= [[As,SD,JK],[LO,WQ],[A]]

Answer 1

我会使用 groupby() 的组合和 .tolist() .但是，您需要先规范化您的 col2 :

df['col2'] = df['col2'].str.lower()
output = df.groupby('col2')['col1'].apply(list).tolist()

这输出:

[['IU'], ['As', 'SD', 'JK'], ['LO', 'WQ']]