python - reduce 以空集为初始值

Question

我有一个列表列表，我想用所有子列表中存在的元素构造一个集合..

示例:a = [[1,2],[2,3]] 应该给出 set([1,2,3])

我试过 reduce(lambda x,y:x.update(y),a,set([])) 但它引发了 AttributeError: 'NoneType' object has no attribute 'update'

谁能告诉我如何使用 reduce 函数做到这一点？

Answer 1

根据要求:

>>> a = [[1,2],[2,3]]
>>> reduce(lambda s, elems: s.union(elems), a, set())
set([1, 2, 3])

另一种方式只是为了好玩:

>>> from itertools import chain
>>> set(chain.from_iterable(a))
set([1, 2, 3])

还有一个很酷:

>>> set.union(set(), *a)
set([1, 2, 3])