Как использовать понимание списка с помощью метода .extend list?

Question

Например, этот фрагмент кода:

out = [] 
for foo in foo_list: 
    out.extend(get_bar_list(foo)) # get_bar_list return list with some data 
return out 

Как сократить этот код, используя список понимание?

Answer 1

Вы можете использовать выражение генератора и потребляют его itertools.chain:

from itertools import chain 

out = list(chain.from_iterable(get_bar_list(foo) for foo in foo_list)) 

Answer 2

Если длина списков невелика, вы можете сделать это с суммированием выражения генератора вместо:

sum((get_bar_list(foo) for foo in foo_list), []) 

nested list comprehension является более эффективным, но менее читаемым.

Answer 3

Вы можете использовать вложенную-список понимание:

out = [foo for sub_item in foo_list for foo in get_bar_list(sub_item)] 

FWIW, у меня всегда есть трудное время вспомнить точно, какие вещи подходят для вложенных списков и поэтому я обычно предпочитаю использовать itertools.chain (как упоминалось в ответе Моисея). Тем не менее, если вы действительно любите вложенные список-постижения, порядок такой же, как вы бы столкнуться с ними в нормальном цикле (с переменной внутренней петлей для первого выражения в списке понимании):

for sub_item in foo_list: 
    for foo in get_bar_list(sub_item): 
     out.append(foo)