[프로그래머스] level1 : 내적

 

파이썬이랑 친해지기

두개의 iteratable한 요소가 존재할 때 zip 함수를 이용하여 튜플 형태로 동시에 순회할 수 있다.

a = [1, 2, 3, 4]
b = ["a", "b", "c", "d"]

for tp in zip(a, b):
    print(tp)
    
# (1, 'a')
# (2, 'b')
# (3, 'c')
# (4, 'd')

예를 들어 위처럼 zip() 함수에 두 리스트를 넣으면 튜플 형태의 iterator를 반환한다.

 

a = [1, 2, 3]
b = ["a", "b", "c", "d"]

for tp in zip(a, b):
    print(tp)
    
# (1, 'a')
# (2, 'b')
# (3, 'c')

만약 두 요소의 개수가 다르다면 더 적은 개수를 기준으로 튜플이 생성된다.

 

코드

def solution(a, b):
    return sum([i * j for i, j in zip(a,b)])