我有一个包含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]
谢谢。
答案 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]