Python을 사용하여 배열 목록에서 0과 1을 분리하시겠습니까?

List Comprehension은 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]