All geek questions in one place

How to make a nested dictionary from list items?

I have a list of lists like:

a = [
     ['A', 'America', 'LA', '1123', '2014-05-01', [('A', '211'), ('AD', '398')]],
     ['D', 'America', 'LA', '1135', '2014-05-01', [('A', '211'), ('AD', '398')]],
     ['I', 'America', 'SF', '1145', '2014-05-01', [('A', '211'), ('AD', '398')]],
     ['A', 'England', 'LND', '3564', '2014-05-01', [('A', '211'), ('AD', '398')]],
     ['D', 'Dubai', 'DUB', '9990', '2014-05-01', [('A', '211'), ('AD', '398')]],
     ['D', 'Dubai', 'DUX', '9670', '2014-05-01', [('A', '211'), ('AD', '398')]],
     ['I', 'Dubai', 'DUB', '9800', '2014-05-01', [('A', '211'), ('AD', '398')]],
    ]

And I want a nested dict:

d = {
     'America': {
                 'LA': {
                        '1123'  : ['A', '2014-05-01', [('A', '211'), ('AD', '398')]],
                        '1135': ['D', '2014-05-01', [('A', '211'), ('AD', '398')]]
                       },
                 'SF': {
                        '1145': ['I', '2014-05-01', [('A', '211'), ('AD', '398')]]
                       }
                 },
     'England': {
                 'LND': {
                        '3564': ['A', '2014-05-01', [('A', '211'), ('AD', '398')]]
                        }

                 },
     'Dubai': {
                 'DUB': {
                        '9990': ['D', '2014-05-01', [('A', '211'), ('AD', '398')]],
                        '9800': ['I', '2014-05-01', [('A', '211'), ('AD', '398')]]
                        },
                 'DUX': {
                        '9670': ['D', '2014-05-01', [('A', '211'), ('AD', '398')]]
                       }
                 }
    }

I tried to do this, but still can not do it, please check my code and help me!

pd = defaultdict(dict)
for row in a:
    country = defaultdict(dict)
    for c in a:
        dest = defaultdict(list)
        for d in a:
            if c[2] == d[2]:
                dest[d[3]].append([d[1], d[0],d[4],d[5],d[6], d[7]])
            else:
                continue
        country[c[2]] = dest
    pd[row[1]] = country
+4
source share
5 answers
from operator import itemgetter
from itertools import groupby
getter, d = itemgetter(0, 4, 5), {}
for key, grp in groupby(sorted(a, key=lambda x: x[3]), key=lambda x: x[3]):
    for item in grp:
        d.setdefault(item[1], {}).setdefault(
            item[2], {})[item[3]] = list(getter(item))
print d

Output

{'America': {'LA': {'1123': ['A',
                             '2014-05-01',
                             [('A', '211'), ('AD', '398')]],
                    '1135': ['D',
                             '2014-05-01',
                             [('A', '211'), ('AD', '398')]]},
             'SF': {'1145': ['I',
                             '2014-05-01',
                             [('A', '211'), ('AD', '398')]]}},
 'Dubai': {'DUB': {'9800': ['I',
                            '2014-05-01',
                            [('A', '211'), ('AD', '398')]],
                   '9990': ['D',
                            '2014-05-01',
                            [('A', '211'), ('AD', '398')]]},
           'DUX': {'9670': ['D',
                            '2014-05-01',
                            [('A', '211'), ('AD', '398')]]}},
 'England': {'LND': {'3564': ['A',
                              '2014-05-01',
                              [('A', '211'), ('AD', '398')]]}}}
+3
source

Try the following:

pd = dict()
for (k4, k1, k2, k3, k5, k6) in a:
    pd.setdefault(k1, dict())
    pd[k1].setdefault(k2, dict())
    pd[k1][k2].setdefault(k3, dict())
    pd[k1][k2][k3] = [k4, k5, k6]

Conclusion:

{'America': {'LA': {'1123': ['A', '2014-05-01', [('A', '211'), ('AD', '398')]],
   '1135': ['D', '2014-05-01', [('A', '211'), ('AD', '398')]]},
  'SF': {'1145': ['I', '2014-05-01', [('A', '211'), ('AD', '398')]]}},
 'Dubai': {'DUB': {'9800': ['I', '2014-05-01', [('A', '211'), ('AD', '398')]],
   '9990': ['D', '2014-05-01', [('A', '211'), ('AD', '398')]]},
  'DUX': {'9670': ['D', '2014-05-01', [('A', '211'), ('AD', '398')]]}},
 'England': {'LND': {'3564': ['A',
    '2014-05-01',
    [('A', '211'), ('AD', '398')]]}}}
+2
source

defaultdicts, for a.

from collections import defaultdict

out = defaultdict(lambda : defaultdict(lambda :defaultdict(dict)))

for item in a:
    out[item[1]][item[2]][item[3]] = [item[0]] + item[4:]

print out == d # Comparing it to the fixed expected output.
print out

:

True

defaultdict(<function <lambda> at 0xadcd70>, 
{'Dubai': defaultdict(<function <lambda> at 0x106a398>, 
         {'DUX': defaultdict(<type 'dict'>,
                {'9670': ['D', '2014-05-01', [('A', '211'), ('AD', '398')]]}),
          'DUB': defaultdict(<type 'dict'>,
                {'9990': ['D', '2014-05-01', [('A', '211'), ('AD', '398')]],
                '9800': ['I', '2014-05-01', [('A', '211'), ('AD', '398')]]})}),
'America': defaultdict(<function <lambda> at 0x1066c08>,
         {'SF': defaultdict(<type 'dict'>,
                {'1145': ['I', '2014-05-01', [('A', '211'), ('AD', '398')]]}),
          'LA': defaultdict(<type 'dict'>,
                {'1123': ['A', '2014-05-01', [('A', '211'), ('AD', '398')]],
                 '1135': ['D', '2014-05-01', [('A', '211'), ('AD', '398')]]})}),
'England': defaultdict(<function <lambda> at 0x106a320>,
         {'LND': defaultdict(<type 'dict'>,
                {'3564': ['A', '2014-05-01', [('A', '211'), ('AD', '398')]]})})})

reduce() operator.itemgetter:

from collections import defaultdict
from operator import itemgetter

out = defaultdict(lambda : defaultdict(lambda :defaultdict(dict)))

indices = (1, 2, 3)
rest = [i for i in range(len(a[0])) if i not in indices]
key_getter = itemgetter(*indices)
value_getter = itemgetter(*rest)


for item in a:
    keys = key_getter(item)
    dct = reduce(lambda x, y: x[y], keys[:-1], out) #fetch the innermost dict
    dct[keys[-1]] = list(value_getter(item))
+2

:

Traceback (most recent call last):
  File "<pyshell#12>", line 7, in <module>
    dest[d[3]].append([d[1], d[0],d[4],d[5],d[6], d[7]])
IndexError: list index out of range

? d, :

['A', 'England', 'LND', '3564', '2014-05-01', [('A', '211'), ('AD', '398')]]
#0    1          2      3       4             5

d[6] d[7] .

:

dest[d[3]].append([d[1], d[0], d[4], d[5]])
0

...

from itertools import groupby

f1 = lambda x: x[1]
f2 = lambda x: x[2]
f3 = lambda x: x[3]

b = sorted(a, key=lambda x: (x[3], x[2], x[1]))
d = {}
for i in groupby(b, f1):
    d[i[0]] = {}
    for j in groupby(i[1], f2):
        d[i[0]][j[0]] = {}
        for k in groupby(j[1], f3):
            d[i[0]][j[0]][k[0]] = []
            for l in k[1]:
                d[i[0]][j[0]][k[0]].append(list(l)[0])
                d[i[0]][j[0]][k[0]].append(list(l)[4])
                d[i[0]][j[0]][k[0]].append(list(l)[5])

:

>>> d
{'America': {'LA': {'1123': ['A', '2014-05-01', [('A', '211'), ('AD', '398')]],
   '1135': ['D', '2014-05-01', [('A', '211'), ('AD', '398')]]},
  'SF': {'1145': ['I', '2014-05-01', [('A', '211'), ('AD', '398')]]}},
 'Dubai': {'DUB': {'9800': ['I', '2014-05-01', [('A', '211'), ('AD', '398')]],
   '9990': ['D', '2014-05-01', [('A', '211'), ('AD', '398')]]},
  'DUX': {'9670': ['D', '2014-05-01', [('A', '211'), ('AD', '398')]]}},
 'England': {'LND': {'3564': ['A',
    '2014-05-01',
    [('A', '211'), ('AD', '398')]]}}}
0

Source: https://habr.com/ru/post/1538586/

More articles:


All Articles