从列表中取出一个字典

时间:2017-06-13 21:34:46

标签: python dictionary

我从我的数据库中获取了一些列表:

print(liste)
[('VEST', 'MODEL-1', 'BEIGE', 'M'), 
 ('VEST', 'MODEL-1', 'BEIGE', 'S'), 
 ('JACKET', 'MOD-1', 'GREEN', 'S'), 
 ('JACKET', 'MOD-1', 'GREEN', 'M'), 
 ('JACKET', 'MOD-2', 'BLACK', 'L'), 
 ('JACKET', 'MOD-2', 'BLACK', 'XL')]

现在我需要创建一个像这样的字典:

mydict = {'VEST' : {'MODEL-1' : {'BEIGE' : ('S', 'M')}},
          'JACKET' : ({'MOD-1' : {'GREEN' : ('S', 'M')}},
                      {'MOD-2' : {'BLACK' : ('L', 'XL')}})}

我会在像json这样的tkinter中使用它。我需要选择类似的东西:

print(mydict['JACKET'][1]['MOD-2']['BLACK'][0])
L
# Or
print(mydict['VEST']['MODEL-1']['BEIGE'][0])
S

如何制作字典?我厌倦了尝试不同的方式。如果我能得到好的建议,我将非常高兴

2 个答案:

答案 0 :(得分:4)

您所展示的内容并不十分一致,但我认为您可能会因使用collections.defaultdict而受益:

In [1]: from collections import defaultdict

In [2]: liste = [('VEST', 'MODEL-1', 'BEIGE', 'M'), 
   ...:  ('VEST', 'MODEL-1', 'BEIGE', 'S'), 
   ...:  ('JACKET', 'MOD-1', 'GREEN', 'S'), 
   ...:  ('JACKET', 'MOD-1', 'GREEN', 'M'), 
   ...:  ('JACKET', 'MOD-2', 'BLACK', 'L'), 
   ...:  ('JACKET', 'MOD-2', 'BLACK', 'XL')]

In [3]: mydict = defaultdict(lambda: defaultdict(lambda: defaultdict(list)))

In [4]: for kind, model, color, size in liste:
   ...:     mydict[kind][model][color].append(size)
   ...:     

In [5]: mydict['VEST']['MODEL-1']['BEIGE']
Out[5]: ['M', 'S']

defaultdict构造函数将“工厂”函数作为参数,即,在没有参数的情况下调用的可调用函数来替换缺失值。例如,当您尝试访问每个丢失的密钥时,defaultdict(list)将有一个空列表。

在这种情况下,我创建了一个嵌套的defaultdict对象,为每个键创建另一个defaultdict。内部也创建defaultdict,它也使用defaultdict作为工厂。这反映了您的数据层次结构,并为丢失的密钥保存了多个检查或异常处理。

答案 1 :(得分:0)

您可以尝试@LevLevitsky所展示的defaultdict,或者您可以尝试这种可行的方式,尽管这种方式很复杂:

result = {}
for i in liste:
    if i[0] not in result.keys():
            result[i[0]] = {i[1]: {i[2]: tuple(i[3:4])}}
    else:
            if i[1] not in result[i[0]].keys():
                   result[i[0]][i[1]] = {i[2]: tuple(i[3:4])}
            else:
                   if i[2] not in result[i[0]][i[1]].keys():
                            result[i[0]][i[1]][i[2]] = tuple(i[3:4])
                   else:
                            result[i[0]][i[1]][i[2]]+=tuple(i[3:4])

print result #{'JACKET': {'MOD-2': {'BLACK': ('L', 'XL')}, 'MOD-1': {'GREEN': ('S', 'M')}}, 'VEST': {'MODEL-1': {'BEIGE': ('M', 'S')}}}