[python] zip 함수, zip 함수의 응용 (feat. 행만 리스트에 저장하는 방법)

 두개의 데이터가 있으면 서로 맞물리게 다른 데이터를 만들 수 있는 방법은 없을까?

 

zip 함수를 이용하면 두개의 데이터를 맞물리게한 새로운 데이터를 만들어 낼 수 있다.

a = [0,1,2,3,4,5]
b = ['a','b','c','d','e','f']

print(list(zip(a,b)))

# 출력 : [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd'), (4, 'e'), (5, 'f')]

 

 

만약 리스트의 크기(길이)가 다르면 어떻게 될까?

a = [0,1,2,3,4,5]
b = ['a','b','c']

print(list(zip(a,b)))

# 출력 : [(0, 'a'), (1, 'b'), (2, 'c')]

위와 같이 최소의 길이에 맞춰서 새로운 데이터가 형성된다. 실제로 ai, 빅데이터, 데이터베이스 등등 많은 데이터를 다루는 개발자들은 zip함수를 자주 다루어 보았을 것 이다. 실제로 알고리즘 문제를 수행하다보면 종종 사용된다. 

이제 zip 함수를 어떻게 사용하는지 예시를 보자

 

 

- zip() 함수의 응용

1. 2차원 리스트(행렬)의 행부분만 추출해서 새로운 데이터 생성

# input : 
# 0 5 4 5 4 5 4 
# 4 0 5 1 2 3 4 
# 9 8 0 1 2 3 1 
# 9 9 9 0 9 9 9 
# 1 1 1 1 0 1 1 
# 8 7 6 5 4 0 3 
# 9 1 9 1 9 1 0 

import sys

n = 7
arr = [list(map(int,sys.stdin.readline().split())) for _ in range(n)]
print(list(zip(*arr)))

# *arr : [0, 5, 4, 5, 4, 5, 4] [4, 0, 5, 1, 2, 3, 4] ...... [9, 1, 9, 1, 9, 1, 0]

출력 해보면 어떻게 될까? 한번 돌려보길 추천한다. 

참고로 * 는 asterisk 라고 해서 리스트를 unpack 해주는 기능을 한다. (더 다양한 기능이 있지만 여기까지만 설명하겠다.)

ex) *[[1,2],[3,4],[5,6]] 하게 되면 [1,2]와 [3,4]와 [5,6]으로 unpack 됩니다. 

 

출력하게 되면 아래와 같이 나온다.

# 출력 :
# [(0, 4, 9, 9, 1, 8, 9), (5, 0, 8, 9, 1, 7, 1), (4, 5, 0, 9, 1, 6, 9), (5, 1, 1, 0, 1, 5, 1), (4, 2, 2, 9, 0, 4, 9), (5, 3, 3, 9, 1, 0, 1), (4, 4, 1, 9, 1, 3, 0)]

 

2. unzip

맞물리게 만들어둔 데이터를 원상복구하는 방법이다.

a = [0,1,2,3,4,5]
b = ['a','b','c']

c = list(zip(a,b))
print(c)
# 출력 : [(0, 'a'), (1, 'b'), (2, 'c')]


# unzip 
print(list(zip(*c)))
# 출력 : [(0, 1, 2), ('a', 'b', 'c')]

 

맞물리게 만들어둔 데이터를 원상복구하는 방법이다.