使用Python分隔数组列表中的0和1？

列表推导是Python中一种流行的技术。在这里，我们使用这种技术。我们创建了一个用户输入数组，并且数组元素应以随机顺序为0和1。然后将左侧的0和右侧的1分开。我们遍历数组并分离两个不同的列表，一个包含0，另一个包含1，然后连接两个列表。

示例

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]