python - 如何在值之后命名 python 变量？

Question

假设我有一个数据框:

     X    Y
1   'a'  ('good','pretty')
2   'b'  ('bad','ugly')
3   'z'  ('ok','soso')

我需要创建三个不同的列表:a 在不同的行中有good 和pretty； b 在不同的行中有坏的和丑陋的等等......

如何通过自动从 X 列获取值来创建这些列表变量来实现此目的？

这是代码:

list1 = ['good, pretty','a']
list2 = ['bad, ugly','b']
list3 = ['ok, soso','c']
test = pd.DataFrame([list1, list2, list3], columns=['desc', 'item'])
test

数据框看起来像:

           desc item
0  good, pretty    a
1     bad, ugly    b
2      ok, soso    c

我需要最终的列表如下:

 {'a':     ('good','pretty'),
  'b':     ('bad','ugly'),
  'c':     ('ok','soso')}

Answer 1

在 groupby 上使用字典理解。

>>> {idx: group['desc'].tolist() 
     for idx, group in test.groupby('item')}

{'a': ['good, pretty'], 'b': ['bad, ugly'], 'c': ['ok, soso']}