将列表元素插入列表列表的Pythonic方法?

时间:2013-09-06 17:43:46

标签: python list

我们有,例如:

nodes = [[1, 2],[3, 4]] 
thelist = [[5, 6], [7, 8]]

我如何编码,所以列表将是:

[[1, 2],[3, 4],[5, 6],[7, 8]]

我知道该怎么做,但我想要一种优雅的蟒蛇方式。

我的尝试:

for node in nodes:
    thelist.insert(0, node)

我想应该有更多的pythonic方法来做到这一点。

编辑:订单在某种程度上很重要(这就是我尝试在索引0处插入的原因)。

3 个答案:

答案 0 :(得分:11)

只需将它们添加到一起:

In [11]: nodes + thelist
Out[11]: [[1, 2], [3, 4], [5, 6], [7, 8]]

您还可以使用extend(修改节点):

In [12]: nodes.extend(thelist)

In [13]: nodes
Out[13]: [[1, 2], [3, 4], [5, 6], [7, 8]]

答案 1 :(得分:8)

您可以指定切片thelist[:0]以在开头插入元素:

nodes = [[1, 2],[3, 4]]
thelist = [[5, 6], [7, 8]]
thelist[:0] = nodes
# thelist is now [[1, 2], [3, 4], [5, 6], [7, 8]]

请参阅lots of useful ways to manipulate lists的Python教程。

答案 2 :(得分:5)

或者,如果某种顺序很重要,或者只是一次只能获取一项,那么您可以使用heapq.merge

import heapq

nodes = [[1, 2],[3, 4]]
thelist = [[5, 6], [7, 8]] 
res = list(heapq.merge(nodes, thelist))
# [[1, 2], [3, 4], [5, 6], [7, 8]]

nodes = [[1, 2], [5,6]]
thelist = [[3, 4], [7, 8]]    
res = list(heapq.merge(nodes, thelist))
# [[1, 2], [3, 4], [5, 6], [7, 8]]

或者,只需使用:

for heapq.merge(nodes, thelist):

请注意,订单可能与itertools.chain不同。