Tách số 0 và số 1 trong danh sách mảng bằng Python?

Tổng hợp danh sách là một kỹ thuật phổ biến trong Python. Ở đây chúng tôi sử dụng kỹ thuật này. Chúng tôi tạo mảng đầu vào của người dùng và phần tử mảng phải là 0 và 1 theo thứ tự ngẫu nhiên. Sau đó, tách riêng 0 ở bên trái và 1 ở bên phải. Chúng tôi duyệt qua mảng và tách hai danh sách khác nhau, một danh sách chứa 0 và một danh sách khác chứa 1, sau đó nối hai danh sách.

Ví dụ

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

Thuật toán

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.

Mã mẫu

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

Đầu ra

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]