试图从 2 个列表创建新字典 - 1 个元组，1 个用于字典键

Question

我有 2 个列表。 a是一个元组列表，我想将其转换为一个字典，该字典将 'a' 值保留为列表 - 使用第二个列表作为字典键。 我尝试使用a's第一组作为元组列表中的“索引”值来与b列表中的“键”值结合。 这就是我所拥有的：

a = [(0, ['Potato'], [8]),
     (0, ['Tomato'], [2]),
     (0, ['Tomato'], [2]),
     (0, ['Potato'], [6]),
     (0, ['Potato'], [12]),
     (0, ['Potato'], [12]),
     (0, nan, nan),
     (1, [], []),
     (1, [], [])]

b = [('foo', 123), ('bar', 456)]

这就是我想要得到的：

newDict = {'foo' : [(['Potato'], ['Tomato'], ['Tomato'], ['Potato'], ['Potato'], ['Potato'], nan), ([8], [2], [2], [6], [12], [12], nan)], 
           'bar' : [([], []),([],[])]}

我尝试通过各种 for 循环进行枚举，解压缩元组。

Answer 1

您可以使用 group by (from itertools) 根据第一个条目对a的元组进行分组，然后将其压缩到b以将 'foo' 和 'bar' 与相应的 0 和 1 组配对：

a = [(0, ['Potato'], [8]),
     (0, ['Tomato'], [2]),
     (0, ['Tomato'], [2]),
     (0, ['Potato'], [6]),
     (0, ['Potato'], [12]),
     (0, ['Potato'], [12]),
     (0, 'nan', 'nan'),
     (1, [], []),
     (1, [], [])]

b = [('foo', 123), ('bar', 456)]


from itertools import groupby

r = {tb[0]:list(zip(*ta))[1:] for tb,(_,ta) in zip(b,groupby(a,lambda t:t[0]))}

print(r)
{'foo': 
  [(['Potato'], ['Tomato'], ['Tomato'], ['Potato'], ['Potato'], ['Potato'], 'nan'), 
   ([8], [2], [2], [6], [12], [12], 'nan')], 
 'bar': 
  [([], []), ([], [])]}