调用返回列表的函数的生成器表达式

Question

我有一个返回列表的函数，我想在使用压缩生成器表达式（或任何漂亮而紧凑的表达式）调用列表时合并该函数的输出

假设我有一个def foo(bar):，其中bar是整数，它经过一些疯狂的复杂计算后返回一个列表。你知道吗

foo(1)=[9,1,5]
foo(2)=[1]
foo(3)=[7,1]

arr=[1,2,3]

我如何在一行代码之后获得arr=[9,1,5,1,7,1]？你知道吗

arr=[foo(x) for x in arr]给了我[[9,1,5],[1],[7,1]]
，我不想再写一行在列表中列出列表。你知道吗

Answer 1

from itertools import chain
result = list(chain.from_iterable(foo(x) for x in arr))