2 つのリストがあります
[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]
1 つのオプションは、 と の組み合わせを使用すること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),())