2つの配列の要素を並び替えを一行で(短く)書きたい

Question

前提・実現したいこと

以下の処理を実現したいのですが、もっとよい書き方はありますか？

該当のソースコード

a = [1,3]
b = [2,4]
c = []
for a,b in zip(a,b):
    c.append(a)
    c.append(b)
print(c)

#実行結果
[1, 2, 3, 4]

Answer 1

これでどうでしょう。

>>> a = [1,3]
>>> b = [2,4]
>>> [y for x in zip(a,b) for y in x]
[1, 2, 3, 4]

---

質問のコードにはまずいところがあるので、コメントしておきます。

a = [1,3]
b = [2,4]
c = []
for a,b in zip(a,b):  # ここでループする変数にa,bを使ったせいで上書きされる。この処理の後元のa,bが必要になったらどうするのかと
    c.append(a)
    c.append(b)
print(c)

# 追加
# 最後にa,bに代入された3,4が表示される
print(a) # => 3
print(b) # => 4

うっかり名前を被らせると、後でバグの温床になったりします。

Comments

[TH_]

なるほど！一緒にしたら駄目なんですね。

[TH_]

やはりリスト内包表記でいけるんですね

[hayataka2049]

c = []
for x in zip(a,b): # xは(1,2), (3,4),...みたいな感じになる
　　for y in x: # yが1,2になって一回終了、二回目でyが3,4になる
　　　　c.append(y)

と等価ですね。ただ多重forの内包表記は慣れないうちはわかりづらいのが難点ですね（つーか慣れててもたまにわからなくなるけど）

[opyon]

https://gyazo.com/579fb7f6e99779b4b6063bd71f31712a
なるほど！勉強になります。
ぱっと見で理解出来なかったのでfor文で組み立て直したら分かりました。

Answer 2

一応ちゃんとしたの書いておきます‥。

c = [y for x in zip(a, b) for y in x]

勘違いコード

c = sorted(a + b)

Comments

[fuzzball]

あ、並べ替えってソートじゃないのか‥。

[TH_]

すみません。質問内容が曖昧でした(汗)

[fuzzball]

今度は被り‥。

Answer 3

ネストした list を flatton する方法を web で探してみました。

a = [1,3]
b = [2,4]

flatten = lambda x: [z for y in x for z in (flatten(y) if hasattr(y, '__iter__') else (y,))]
flatten(zip(a,b))

sum(zip(a, b), ())

import numpy as np
np.array(list(zip(a, b))).flatten()

from itertools import chain
list(chain.from_iterable(zip(a, b)))

from functools import reduce
reduce(lambda a, b: a + b, zip(a, b))

[y for x in zip(a, b) for y in x]

実行例

参考情報

• Python 3 で flatten する方法いろいろ

https://qiita.com/hoto17296/items/e1f80fef8536a0e5e7db

• Flatten a list of lists in one line in Python

https://coderwall.com/p/rcmaea/flatten-a-list-of-lists-in-one-line-in-python

Answer 4

ご提示のコードが最もシンプルだと思いますが、参考までに。

>>> from itertools import chain
>>>
>>> a = list('123')
>>> b = list('abc')
>>>
>>> c = list(chain.from_iterable(zip(a, b)))
>>> c
['1', 'a', '2', 'b', '3', 'c']
>>>
>>> c = list(chain(*zip(a, b)))
>>> c
['1', 'a', '2', 'b', '3', 'c']

あるいはサードパーティの more_itertools.interleave を使います。

>>> import more_itertools
>>>
>>> a = list('123')
>>> b = list('abc')
>>>
>>> c = list(more_itertools.interleave(a, b))
>>> c
['1', 'a', '2', 'b', '3', 'c']

Comments

[TH_]

ありがとうございます。intertoolsを使えばよいのですね。

[LouiS0616]

書いておいて難ですが、hayataka2049さんの方法の方がシンプルなように思います。

Answer 5

a = [1,3]
b = [2,4]
c = sum(zip(a, b), tuple())