[Programmers] 내적

내적

월간 코드 챌린지 1

level 1

 

python3

def solution(a, b):
    answer = 0
    for i in range(len(a)):
        answer += a[i]*b[i]
    return answer

처음에는 [x * y for x in a for y in b]로 해봤으나, 내적이 아니라 약간 이중 for문 처럼 나와버려서 당황..

사실 순열을 array comprehension으로 가능하다는 어마어마한 유틸성에 놀랐다...

 

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

zip 함수를 이용해서 a배열과 b배열의 각 원소를 튜플로 만든 후, 해당 튜플에 대해서 곱 연산을 한 뒤 sum 을 해주는 것도 가능하다.