Можно ли создать список всех ключей-членов из dicts dicts, используя понимание списка?

Question

Скажем, у меня есть Dict из dicts:

foo = { 
    'category': { 
     'key1': 'bar', 
     'key2': 'bar', 
     'key3': 'bar', 
    }, 
    'category2': { 
     'key4': 'bar', 
     'key5': 'bar', 
    }, 
} 

Чтобы получить единый список всех ключей в странах-dicts, у меня есть функция следующим образом:

def _make_list(): 
    baz = list() 
    for key,val in foo.items(): 
     baz += list(val.keys()) 
    return baz 

Сформированный список выглядит : ['key1', 'key2', 'key3', 'key4', 'key5', ].

Это достаточно просто, и это работает, но мне интересно: есть ли способ сделать это с пониманием списка с одним слоем? Ключи участников dicts всегда будут уникальными.

Answer 1

Вот один из способов сделать это:

>>> [k for d in foo.values() for k in d] 
['key1', 'key2', 'key3', 'key4', 'key5'] 

Answer 2

подход будет использовать itertools.chain():

import itertools 

[k for k in itertools.chain(*(d.keys() for d in foo.values()))] 

Если то, что вы хотите, это просто одна строка кода, а не обязательно список понимание вы можете также попробовать (упомянутые @Duncan):

list(itertools.chain(*foo.values())) 

Выход:

>>> [k for k in itertools.chain(*(d.keys() for d in foo.values()))] 
['key3', 'key2', 'key1', 'key5', 'key4'] 
>>> 
>>> list(itertools.chain(*foo.values())) 
['key3', 'key2', 'key1', 'key5', 'key4']