List Comprehension is a popular technique in Python. Here we use this technique. We create a user input array and array element should be 0’s and 1’s in random order. Then segregate 0’s on left side and 1’s on right side. We traverse the array and separate two different list, one contains 0’s and another contains 1’s, then concatenate two list.
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]