如何使用列表理解来扩展 python 中的列表？

Question

我在 Python 方面没有经验，我经常编写（简化的）如下所示的代码：

accumulationList = []
for x in originalList:
    y = doSomething(x)
    accumulationList.append(y)
return accumulationList

然后在测试通过后，我重构为

return [doSomething(x) for x in originalList]

但假设结果有点不同，我的循环如下所示：

accumulationList = []
for x in originalList:
    y = doSomething(x)
    accumulationList.extend(y)
return accumulationList

其中 doSomething list 返回一个列表。完成这个任务最Pythonic的方法是什么？显然，前面的列表理解会给出一个列表列表。

Answer 1

使用列表理解更简单、更清晰：

[y for x in originalList for y in doSomething(x)]

Answer 2

你的意思是这样的吗？

accumulationList = []
for x in originalList:
    accumulationList.extend(doSomething(x))
return accumulationList

或更短的代码（但不是最佳的）：

return sum((doSomething(x) for x in originalList), [])

或者相同：

return sum(map(doSomething, originalList), [])

感谢@eyquem的提示（如果使用Python 2.x）：

import itertools as it

return sum(it.imap(doSomething, originalList), [])

Answer 3

我认为涉及 add 或 iadd 的答案在二次时间中运行，这可能不好。我会尝试：

from itertools import chain
accumulation_list = list(chain.from_iterable(doSomething(x) for x in originalList))

Answer 4

Python 的就地添加运算符（+=，在 operator 模块中以 iadd 形式提供）相当于 .extend对于列表。将其与reduce配对以获得您想要的。

import operator

reduce(operator.iadd, (doSomething(x) for x in originalList)
, accumulation_list)

Answer 5

我认为这种情况没有特殊的语法。但是你可以使 for 循环更短：

accumulationList += doSomething(x)

如果你坚持，你可以使用函数式编程来展平列表：

result = reduce(lambda a,b: a+b, [[i,i*2] for i in range(3)])

但我不会称其为 pythonic，我认为它比 for 循环更难阅读。

Answer 6

功能上，您可以使用itertools.chain 与 地图。对于输入列表 L：

res = list(chain.from_iterable(map(doSomething, L)))

如果您需要迭代器，只需删除 list 调用即可。这是一个演示：

def doSomething(value):
    return [value * i for i in range(1, 4)]

def original(originalList):
    accumulationList = []
    for x in originalList:
        y = doSomething(x)
        accumulationList.extend(y)
    return accumulationList

def new(L):
    return list(chain.from_iterable(map(doSomething, L)))

x = [1, 2, 3]

assert original(x) == new(x)

print(new(x))

[1, 2, 3, 2, 4, 6, 3, 6, 9]