合并切片列表

时间:2012-04-18 17:58:51

标签: python list

我有一个包含N个元素的列表,我使用某个步骤对其进行切片,比方说3:

slice0 = text[0::3]
slice1 = text[1::3]
slice2 = text[2::3]

在分开处理之后,现在我需要将它们合并回原始列表中的相同位置。是否有类似(简单)的方法来做到这一点?

示例:

L = [1,2,3,4,5,6] -> L0 = [1,4], L1 = [2,5], L2 = [3,6]

然后进行一些处理(比如将每个列表分别乘以1,2和3:

L0 = [1,4], L1 = [4,10], L2 = [9,18] 

将它们合并回原来的位置

L = [1,4,9,4,10,18]

谢谢。

4 个答案:

答案 0 :(得分:5)

您可以使用zip()功能将它们重新组合在一起:

>>> l0 = [1,4]; l1 = [4,10]; l2 = [9,18]
>>> zip(l0, l1, l2)
[(1, 4, 9), (4, 10, 18)]
>>> [x for t in zip(l0, l1, l2) for x in t]
[1, 4, 9, 4, 10, 18]

或使用itertools.chain

>>> from itertools import chain
>>> list(chain(*zip(l0, l1, l2)))
[1, 4, 9, 4, 10, 18]

在Python 3中,zip是一个生成器函数,itertools.chain.from_iterable可能是首选的,正如其他人已经指出的那样。

答案 1 :(得分:4)

>>> L0 = [1,4]
>>> L1 = [4,10]
>>> L2 = [9,18]
>>> [x for zipped in zip(L0, L1, L2) for x in zipped]
[1, 4, 9, 4, 10, 18]

或使用itertools.chain的替代方案:

>>> from itertools import chain
>>> list(chain.from_iterable(zip(L0, L1, L2)))
[1, 4, 9, 4, 10, 18]

答案 2 :(得分:2)

>>> from itertools import chain
>>> l0, l1, l2, = [1,4], [4,10], [9,18]
>>> list(chain.from_iterable(zip(l0,l1,l2)))
[1, 4, 9, 4, 10, 18]

答案 3 :(得分:0)

L0 = [1,4]
L1 = [4,10] 
L2 = [9,18]

#returns a tuple

my_data = sum(zip(L0,L1,L2),())

#or if you need a list

my_data = list(sum(zip(L0,L1,L2),()))

元组:

>>> my_data
(1, 4, 9, 4, 10, 18)

列表:

>>> my_data
[1, 4, 9, 4, 10, 18]