Comprensión de listas de Python | Separe 0’s y 1’s en una lista de array

Se le da una array de 0 y 1 en orden aleatorio. Separe los 0 en el lado izquierdo y los 1 en el lado derecho de la array.

Ejemplos:

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

Tenemos una solución existente para este problema, consulte Segregar 0 y 1 en un enlace de array. Podemos resolver este problema rápidamente en Python usando List Comprehension . Recorra la lista dada y separe dos listas diferentes, una contiene todos los 0 y otra contiene todos los 1. Ahora concatene ambas listas juntas.

# Function to Segregate 0's and 1's in an array list
  
def segregate(arr):
    res = ([x for x in arr if x==0] + [x for x in arr if x==1])
    print(res)
  
# Driver program
if __name__ == "__main__":
    arr = [0, 1, 0, 1, 0, 0, 1, 1, 1, 0] 
    segregate(arr)

Producción:

[0, 0, 0, 0, 0, 1, 1, 1, 1, 1]