在python中将两个列表拼接在一起的有效方法

Question

我在 python 中有两个嵌套列表，例如：

lst1 = [
       [ [1,2],[3,4] ], [ [5,6,7],[8,9,10] ]
       ]
lst2 = [
       [ [11,12],[13,14] ], [ [15,16,17],[18,19,20] ]
       ]

我希望能够放入与列表的扁平版本相对应的索引，并返回“交叉”列表：

例如index 为 5，输出为新列表

lst3 = [
       [ [1,2],[3,4] ], [ [5,16,17],[18,19,20] ]
       ]

我可以通过执行三重嵌套循环来做到这一点，但我知道在 python 中必须有更快的方法，可能使用迭代器？

编辑：我在问如何更有效地执行以下操作：

lst1 = [
       [ [1,2],[3,4] ], [ [5,6,7],[8,9,10] ]
       ]
lst2 = [
       [ [11,12],[13,14] ], [ [15,16,17],[18,19,20] ]
       ]


lst3 = []

counter = 0
for i in xrange(len(lst1)):
    for j in xrange(len(lst1[i])):
        temp1 = []
        for k in xrange(len(lst1[i][j])):
            temp2 = []
            counter += 1
            if counter > 5:
                temp2.append(lst2[i][j][k])
            else:
                temp2.append(lst1[i][j][k])
            temp1.append(temp2)
        lst3.append(temp1)

print lst1
print lst2
print lst3

lst3 结果：

[[[1], [2]], [[3], [4]], [[5], [16], [17]], [[18], [19], [20]]]

Answer 1

我不确定您所说的“交叉”列表是什么意思，但您似乎正在尝试对列表进行某种regex。但是您还提到您想要一个 扁平化 列表，即使您的输出没有扁平化。

所以，我不确定您要的是什么。但是，如果我仅使用标题来猜测您要查找的内容，那么答案相当简单，使用list comprehension：

>>> lst3 = [x for x in [lst1, lst2]]

这将给出以下结果：

>>> lst3
[[[[1, 2], [3, 4]], [[5, 6, 7], [8, 9, 10]]], [[[11, 12], [13, 14]], [[15, 16, 17], [18, 19, 20]]]]

这是两个列表的组合。不确定它是否是您想要的方式的“组合”，因为可以以多种方式解释它，我无法准确说出您想要什么。