[Python3] 내장 함수 zip 활용 예제 - 2차원 리스트 행과 열 뒤집기

zip 함수

mylist = [ 1,2,3 ]
new_list = [ 40, 50, 60 ]
for i in zip(mylist, new_list):
    print (i)

(1, 40)
(2, 50)
(3, 60)

 

리스트 두 개를 하나로 묶어 출력이 가능하다. zip(자료구조, 자료구조) 형태로 사용한다.

 

물론 해당 자료구조는 iterable이므로 set, list와 같은 자료구조만 가능하다.

 

list1 = [1, 2, 3, 4]
list2 = [100, 120, 30, 300]
list3 = [392, 2, 33, 1]
answer = []

for i, j, k in zip(list1, list2, list3):
   print( i + j + k )

 

위와 같이 2개 뿐만 아니라 그 이상의 여러개를 묶어서도 가능하다. 

 

위의 예제는 list1, list2, list3의 요소들을 하나씩 뽑아 합산하는 예제이다. 493, 124, 66, 305가 출력된다.

 

animals = ['cat', 'dog', 'lion']
sounds = ['meow', 'woof', 'roar']
answer = dict(zip(animals, sounds)) # {'cat': 'meow', 'dog': 'woof', 'lion': 'roar'}

 

또한, 리스트 2개를 zip을 사용하여 묶고, dict로 변환하여 사전 형태(dictionary)로 사용이 가능하다. 

 

기존 dict와 같이 출력하면 {'cat': 'meow', 'dog': 'woof', 'lion': 'roar'} 형태로 출력된다. 

 

 

zip 함수 활용 예제 - 2차원 리스트 행과 열 뒤집기

 

mylist = [[1,2,3,4], [5,6,7,8], [9,10,11,12],[13,14,15,16]]

def solution(mylist):
    answer = []
    s = len(mylist)

    for i in range(s):
        answer.append([])

    for i in range(s):
        for j in range(s):
            answer[i].append(mylist[j][i])

    return answer
    
print (solution(mylist))

 

위와 같은 예제를 zip함수를 사용하여 아주 깔끔하게 작성이 가능하다. 우선 위의 출력은 다음과 같다.

 

[[1, 5, 9, 13], [2, 6, 10, 14], [3, 7, 11, 15], [4, 8, 12, 16]]

 

mylist = [[1,2,3,4], [5,6,7,8], [9,10,11,12],[13,14,15,16]]

def solution(mylist):
    answer = list(map(list, zip(*mylist)))
    return answer

print (solution(mylist))

 

zip함수를 사용하여 깔끔한 소스코드를 작성하였다. answer 리스트를 zip함수를 사용하여 mylist의 각 행의 요소를 묶는다. 이해하기 쉽게

 

 

[1,2,3,4]

[5,6,7,8]

[9,10,11,12]

[13,14,15,16]

 

위에서 *mylist라는 것은 각 행에 대해서 같은 위치에 대해 요소를 가리키게 된다. 따라서 zip함수를 사용하게 되면 처음에는 1, 5, 9, 13의 숫자를 묶는다. 그 다음에는 2, 6, 10, 14를 묶어준다...

 

이렇게 묶인 숫자들을 map형태로 list의 원소로 저장한다.

 

list = [[1, 5, 9, 13], [2, 6, 10, 14], [3, 7, 11, 15], [4, 8, 12, 16]]