我有以下df,并希望创建一个字典clust
作为键和Feature
值列表。我将如何创建desired dict
?
df
Feature clust
0 I_0 2
1 I_1 3
2 I_2 1
3 I_3 1
4 I_4 2
5 N_0 3
6 N_1 3
desired dict
{1: ['I_2','I_3'],
2: ['I_0','I_4'],
3: ['I_1','N_0','N_1']}
干杯:)
首先将汇总列表Series
,然后将其转换为dictionary
:
desired dict = df.groupby('clust')['Feature'].agg(list).to_dict()
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句