[python] zip() 함수

# zip()함수는 다른 그룹의 데이터를 서로 엮어줌
# zip()함수는 여러개의 순회가능한(iterable) 객체를 input으로 받아 
# 순서대로 짝을 지어 튜플의 형태로 접근할 수 있는 반복자(iterator)를 return

nums = [1, 2, 3, 4]
letters = ['a', 'b', 'c']
for pair in zip(nums, letters):
    print(pair)
#(1, 'a')
#(2, 'b')
#(3, 'c')

nums = [1, 2, 3, 4]
letters_1 = ['A', 'B', 'C', 'D']
letters_2 = ['a', 'b', 'c', 'd']
for pair in zip(nums, letters_1, letters_2):
    print(pair)
#(1, 'A', 'a')
#(2, 'B', 'b')
#(3, 'C', 'c')
#(4, 'D', 'd')

# 문자열이 input일 경우
for pair in zip('1234', 'ABCD', 'abcd'):
    print(pair)
#('1', 'A', 'a')
#('2', 'B', 'b')
#('3', 'C', 'c')
#('4', 'D', 'd')

# 엮어놓은 데이터를 다시 해제(unzip)
# unpacking 연산자 *  붙여서 다시 zip() 함수에 input
pairs = list(zip('1234', 'ABCD'))
print(pairs) #[('1', 'A'), ('2', 'B'), ('3', 'C'), ('4', 'D')]
nums, letters = zip(*pairs)
print(nums) #('1', '2', '3', '4')
print(letters) #('A', 'B', 'C', 'D')

#zip()한 객체를 dict형태로 변환
nums = [1, 2, 3]
letters = ['A', 'B', 'C']
print(dict(zip(nums, letters)))
#{1: 'A', 2: 'B', 3: 'C'}

#길이가 안맞을 경우 => 짧은걸 기준으로 생성하고 나머지는 버림
numbers = ["1", "2", "3"]
letters = ['A']
print(list(zip(numbers, letters))) 
# ('1', 'A')]

📒 zip() 활용 - 2차원 배열 뒤집기

arr = ['1234', '1234', '1234', '1234']
print(list(zip(*arr)))  
#[('1', '1', '1', '1'), ('2', '2', '2', '2'), ('3', '3', '3', '3'), ('4', '4', '4', '4')]

arr2 = [[1,2,3,4], [1,2,3,4], [1,2,3,4], [1,2,3,4]]
print(list(zip(*arr2)))
#[(1, 1, 1, 1), (2, 2, 2, 2), (3, 3, 3, 3), (4, 4, 4, 4)]