python - 如何将reduce与列表一起使用

Question

以下有什么问题？：

lss = reduce(list.extend, [[1],[2]], [])

原因：

Traceback (most recent call last):
  File "<pyshell#230>", line 1, in <module>
    lss = reduce(list.extend, [[1],[2]], [])
TypeError: descriptor 'extend' requires a 'list' object but received a 'NoneType'

我不确定NoneType它来自哪里。

Answer 1

试试这个：

lss = reduce(lambda acc, ele : acc + ele, [[1],[2]], [])

lss
> [1, 2]

问题是extend()返回None（这NoneType就是来自的地方），这不适用于您想要做的事情 - 传递给的函数必须返回一个值：到目前为止的累积结果。reduce()

Answer 2

我认为值得注意的是：

sum([[1],[2]], [])

也可以，而且我很确定会比通过 lambda 减少更快。

我很好奇不同方法的速度，所以我做了一些测试：

reduce(lambda a,b:a+b, x, [])               3644.38161492
reduce(list.__add__, x, [])                 3609.44079709
sum(x,[])                                   3526.84987307
y = [];for z in x: y.extend(z)               143.370306969
y = [];map(y.extend,x)                        71.7020270824
y = [None]*400;del y[:];map(y.extend,x)       66.2245891094
list(itertools.chain(*x))                    102.285979986
list(itertools.chain.from_iterable(x))        96.6231369972
[a for b in x for a in b]                    203.764872074

在 PyPy 上（因为，为什么不）

reduce(lambda a,b:a+b, x, [])               4797.5895648
reduce(list.__add__, x, [])                 4794.01214004
sum(x,[])                                   4748.02929902
y = [];for z in x: y.extend(z)                56.9253079891
y = [];map(y.extend,x)                        73.8642170429
y = [None]*400;del y[:];map(y.extend,x)      152.157783031
list(itertools.chain(*x))                    633.854824066
list(itertools.chain.from_iterable(x))       629.917827129
[a for b in x for a in b]                     89.6922459602

x = [[1,2,3,4],[2,3,4,5],[3,4,5,6],[4,5,6,7],[5,6,7,8],[6,7,8,9],[7,8,9,10],[8,9,10,11]]*100

结论：

1. 在 reduce 中使用 lambda 很慢
2. 专门的sum功能比减少更快
3. 添加列表很慢。
4. Python 循环开销很大。

Answer 3

正如 Óscar López 所指出的，list.extend返回None，因此不能与reduce. 除了建议使用该lambda功能，您还可以使用list.__add__with reduce：

>>> reduce(list.__add__, [[1],[2]], [])
[1, 2]

Answer 4

您可能想使用itertools.chain

创建一个迭代器，从第一个迭代器返回元素，直到它用尽，然后继续到下一个迭代器，直到所有的迭代器都用尽。用于将连续序列视为单个序列。相当于：

def chain(*iterables):
    # chain('ABC', 'DEF') --> A B C D E F
    for it in iterables:
        for element in it:
            yield element