Python列表理解 |分隔数组列表中的 0 和 1

你会得到一个随机排列的 0 和 1 数组。将数组左侧的 0 和右侧的 1 分隔开。

例子：

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

我们有针对此问题的现有解决方案，请参阅在数组链接中隔离 0 和 1。我们可以在Python中使用 List Comprehension 快速解决这个问题。遍历给定列表并分离出两个不同的列表，一个包含所有 0，另一个包含所有 1。现在将两个列表连接在一起。

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

输出：

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