Pythonで配列リスト内の0と1を分離する方法

リスト内包表記（List Comprehension）はPythonで広く使われている人気のテクニックです。ここでは、この手法を活用して配列内の0と1を分離してみましょう。

まず、ユーザーから入力を受け取って配列を作成します。配列の要素は0と1がランダムな順序で含まれているものとします。そして、0を配列の左側へ、1を右側へと分離させます。

具体的には、配列を走査しながら要素を2つの異なるリストに振り分けます。片方のリストには0だけを、もう片方のリストには1だけを格納し、最後にその2つのリストを連結することで目的の結果を得られます。

実行例

入力： a=[0,1,1,0,0,1]
出力： [0,0,0,1,1,1]

アルゴリズム

seg0s1s(A)
/* Aはユーザー入力の配列であり、Aの要素は0と1の組み合わせとする */
ステップ1： まず配列全体を走査する。
ステップ2： 配列の各要素を確認する。要素が0であれば左側に、
           1であれば右側に配置する。
ステップ3： 分けた2つのリストを連結する。

サンプルコード

#  配列リスト内の0と1を分離する
#  Segregate 0's and 1's in an array list
def seg0s1s(A):
   n = ([i for i in A if i==0] + [i for i in A if i==1])
   print(n)

# ドライバープログラム
if __name__ == "__main__":
   A=list()
   n=int(input("配列のサイズを入力してください ::"))
   print("数値を入力してください ::")
   for i in range(int(n)):
      k=int(input(""))
      A.append(int(k))
   print("新しい配列リスト ::")    
   seg0s1s(A)

実行結果

配列のサイズを入力してください ::6
数値を入力してください ::
1
0
0
1
1
0
新しい配列リスト ::
[0, 0, 0, 1, 1, 1]

このように、リスト内包表記を使えばわずか数行のコードで0と1を効率的に分離できます。条件に合致する要素を抽出する処理が直感的に書けるため、可読性の高いコードになります。