如何将列表理解与 .extend 列表方法一起使用？ [复制]

Question

比如这个sn-p：

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

您可以使用嵌套列表理解：

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)