Pythonを使用して配列リストの0と1を分離しますか？

リスト内包表記は、Pythonで人気のある手法です。ここでは、この手法を使用します。ユーザー入力配列を作成します。配列要素はランダムな順序で0と1である必要があります。次に、左側の0と右側の1を分離します。配列をトラバースし、2つの異なるリストを分離します。1つは0を含み、もう1つは1を含み、2つのリストを連結します。

例

Input:: a=[0,1,1,0,0,1]
Output::[0,0,0,1,1,1]

アルゴリズム

seg0s1s(A)
/* A is the user input Array and the element of A should be the combination of 0’s and 1’s */
Step 1: First traverse the array.
Step 2: Then check every element of the Array. If the element is 0, then its position is left side and if 1 then it is on the right side of the array.
Step 3: Then concatenate two list.

サンプルコード

#  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)

# Driver program
if __name__ == "__main__":
   A=list()
   n=int(input("Enter the size of the array ::"))
   print("Enter the number ::")
   for i in range(int(n)):
      k=int(input(""))
      A.append(int(k))
   print("The New ArrayList ::")    
   seg0s1s(A)

出力

Enter the size of the array ::6
Enter the number ::
1
0
0
1
1
0
The New ArrayList ::
[0, 0, 0, 1, 1, 1]