[pythonic code] 리스트값을 출력하는 다양한 방법 enumerate() zip()

 

💡enumerate()

enumerate()함수는 리스트의 값을 추출할 때 '인덱스를 붙여' 함께 출력하는 함수이다.

 

>>>seasons = ['Spring', 'Summer', 'Fall', 'Winter']

>>>list(enumerate(seasons))
[(0, 'Spring'), (1, 'Summer'), (2, 'Fall'), (3, 'Winter')]

>>>list(enumerate(seasons, start=1))
[(1, 'Spring'), (2, 'Summer'), (3, 'Fall'), (4, 'Winter')]

enumerate()함수의 start 파라미터로 인덱스의 시작점도 지정할 수 있다.

 

def enumerate(sequence, start=0):
    n = start
    for elem in sequence:
        yield n, elem
        n += 1

 

 

💡zip()

zip() 함수는 1개 이상의 리스트값이 같은 인덱스에 있을 때 병렬로 묶는 함수이다.

>>> lst_a = ['a1', 'a2', 'a3']
>>> lst_b = ['b1', 'b2', 'b3']

>>> for a, b in zip(lst_a, lst_b):
...     print(a, b)
...
a1 b1
a2 b2
a3 b3

>>> for item in zip([1, 2, 3], ['sugar', 'spice', 'everything nice']):
...     print(item)
...
(1, 'sugar')
(2, 'spice')
(3, 'everything nice')

zip함수는 matrix를 전치하는 것과 비슷하다.

 

>>> list(zip(range(3), ['fee', 'fi', 'fo', 'fum']))
[(0, 'fee'), (1, 'fi'), (2, 'fo')]

>>> list(zip(range(3), ['fee', 'fi', 'fo', 'fum'], strict=True))
Traceback (most recent call last):
  ...
ValueError: zip() argument 2 is longer than argument 1

zip() 함수는 같은 인덱스에 있을 때 묶어주기 때문에, 두 개의 list의 length가 같아야하는데, 만약 두 리스트의 length가 다르면 같은 곳 까지만 묶어준다.

하지만, strict 파라미터를 true로 지정하면 두 리스트의 length가 다르면 error가 발생한다.