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]]

标签： pandas

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']]