Python中许多列表的连接

Question

假设我有这样的函数:

def getNeighbors(vertex)

它返回一个顶点列表,这些顶点是给定顶点的邻居.现在我想创建一个包含邻居所有邻居的列表.我这样做:

listOfNeighborsNeighbors = []
for neighborVertex in getNeighbors(vertex):
    listOfNeighborsNeighbors.append(getNeighbors(neighborsVertex))

是否有更多的pythonic方式来做到这一点？

Answer 1

像往常一样,itertools模块包含一个解决方案:

>>> l1=[1, 2, 3]

>>> l2=[4, 5, 6]

>>> l3=[7, 8, 9]

>>> import itertools

>>> list(itertools.chain(l1, l2, l3))
[1, 2, 3, 4, 5, 6, 7, 8, 9]

Answer 2

[x for n in getNeighbors(vertex) for x in getNeighbors(n)]

要么

sum(getNeighbors(n) for n in getNeighbors(vertex), [])

Answer 3

附加列表可以使用+和sum()完成:

>>> c = [[1, 2], [3, 4]]
>>> sum(c, [])
[1, 2, 3, 4]

Answer 4

如果速度很重要,最好使用它:

from operator import iadd
reduce(iadd, (getNeighbors(n) for n in getNeighbors(vertex)))

这段代码的重点是连接整个列表list.extend,列表理解将逐个添加一个项目,就像调用一样list.append.这节省了一些开销,使前者(根据我的测量)大约快三倍.(iadd操作员通常被写为+=和做同样的事情list.extend.)

使用列表推导(Ignacio的第一个解决方案)通常仍然是正确的方式,它更容易阅读.

但绝对避免使用sum(..., []),因为它以二次方运行.对于许多列表(超过一百个左右)来说,这是非常不切实际的.

Answer 5

按速度排序:

list_of_lists = [[x,1] for x in xrange(1000)]

%timeit list(itertools.chain(*list_of_lists))
100000 loops, best of 3: 14.6 µs per loop

%timeit list(itertools.chain.from_iterable(list_of_lists))
10000 loops, best of 3: 60.2 µs per loop

min(timeit.repeat("ll=[];\nfor l in list_of_lists:\n ll.extend(l)", "list_of_lists=[[x,1] for x in xrange(1000)]",repeat=3, number=100))/100.0
9.620904922485351e-05

%timeit [y for z in list_of_lists for y in z]
10000 loops, best of 3: 108 µs per loop

%timeit sum(list_of_lists, [])
100 loops, best of 3: 3.7 ms per loop