[Python function] Python 문법

zip() 함수

---

zip() 함수는 2개 이상의 시퀀스를 짧은 길이를 기준으로 일대일 대응하는 새로운 튜플 시퀀스를 만드는 역할을 한다.

a = [1, 2, 3, 4, 5]
b = [2, 3, 4, 5]
c = [3, 4, 5]
zip(a, b)
# <zip object at 0x105b6d9b0>

파이썬 3+에서는 제너레이터를 리턴한다. 제너레이터에서 실제값을 추출하기 위해서는 list()로 한 번 더 묶어주면 된다.

list(zip(a, b))
# [(1, 2), (2, 3), (3, 4), (4, 5)]
list(zip(a, b, c))
# [(1, 2, 3), (2, 3, 4), (3, 4, 5)]

zip()의 결과 자체는 리스트 시퀀스가 아닌 튜플 시퀀스를 만들기 때문에, 값을 변경하는 게 불가능하다. 불변(Immutable) 객체다.

즉, zip()은 여러 시퀀스에서 동일한 인덱스의 아이템을 순서대로 추출하여 튜플로 만들어 주는 역할을 한다.

아스테리스크(*)

---

파이썬에서 *는 언팩(Unpack)이다. 시퀀스 언패킹 연산자(Sequence Unpacking Operator)로 말 그대로 시퀀스를 풀어헤치는 연산자를 뜻하며, 주로 튜플이나 리스트를 언패킹하는 데 사용한다.

collections.Counter(nums).most.common(k)
# [(1, 3), (2, 2)]

# 언패킹을 했을 때
list(zip(*collections.Counter(nums).most_common(k)))
# [(1, 2), (3, 2)]

# 언패킹을 하지 않았을 때
list(zip(collections.Counter(nums).most_common(k)))
#[((1, 3),), ((2, 2),)]

fruits = ['lemon', 'pear', 'watermelon', 'tomato']
fruits
# ['lemon', 'pear', 'watermelon', 'tomato']

print(*fruits)
# lemon pear watermelon tomato

언패킹뿐만 아니라 함수의 파라미터가 되었을 떄는 반대로 패킹(Packing)도 가능하다.

def f(*params):
    print(params)

f('a', 'b', 'c')
#('a', 'b', 'c')

하나의 파라미터를 받는 함수에 3개의 파라미터를 전달했지만, params 변수 하나로 패킹되어 처리된다.

a, *b = [1, 2, 3, 4]
a
# 1
b
# [2, 3, 4]

*a , b = [1, 2, 3, 4]
a
# [1, 2, 3]
b
# 4

변수의 할당 또한 *로 묶어서 처리할 수 있다. 일반적으로 변수는 값을 하나만 취하지만 *로 처리하게 되면 나머지 모든 값을 취하게 된다. 하나가 아닌 2개를 쓰는 경우도 있다. ** 2개는 키/값 페어를 언패킹하는 데 사용된다.

data_info = {'year': '2020', 'month': '01', 'day': '7'}
new_info = {**data_info, 'day' : '14'}
new_info
# {'year': '2020', 'month': '01', 'day': '14'}

itertools module

---