Python列表理解|在数组列表中分隔0和1-yiteyi-C++库

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

null

例如：

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

推荐：请尝试你的方法 {IDE} 首先，在进入解决方案之前。

我们有解决这个问题的现有方案，请参考在数组中分隔0和1 链接我们可以使用Python快速解决这个问题列表理解。遍历给定的列表并分离出两个不同的列表，一个包含所有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]

文章版权归作者所有，未经允许请勿转载。

THE END

Python