我有两个列表:
[a, b, c] [d, e, f]
我想要:
[a, d, b, e, c, f]
在 Python 中执行此操作的简单方法是什么?
这是一个使用列表推导的非常简单的方法:
>>> lists = [['a', 'b', 'c'], ['d', 'e', 'f']]
>>> [x for t in zip(*lists) for x in t]
['a', 'd', 'b', 'e', 'c', 'f']
或者,如果您将列表作为单独的变量(如其他答案):
[x for t in zip(list_a, list_b) for x in t]
一种选择是使用chain.from_iterable()
和的组合zip()
:
# Python 3:
from itertools import chain
list(chain.from_iterable(zip(list_a, list_b)))
# Python 2:
from itertools import chain, izip
list(chain.from_iterable(izip(list_a, list_b)))
编辑:正如评论中 sr2222 所指出的,如果列表的长度不同,这将无法正常工作。在这种情况下,根据所需的语义,您可能希望使用文档配方部分中的
(更通用的)roundrobin()
函数:itertools
def roundrobin(*iterables):
"roundrobin('ABC', 'D', 'EF') --> A D E B F C"
# Recipe credited to George Sakkis
pending = len(iterables)
nexts = cycle(iter(it).next for it in iterables)
while pending:
try:
for next in nexts:
yield next()
except StopIteration:
pending -= 1
nexts = cycle(islice(nexts, pending))
这个仅适用于 python 2.x,但适用于不同长度的列表:
[y for x in map(None,lis_a,lis_b) for y in x]
您可以使用内置函数做一些简单的事情:
sum(zip(list_a, list_b),())