본문 바로가기
2021/코딩테스트

[프로그래머스] Lv.1 - 내적 (by python)

by 에이도 2020. 12. 3.

처음 내적은 numpy의 dot 기능을 통해 내적을 하려고 했지만, 

type의 형태가 int64이기 때문에 json으로 읽을 수 없다고 합니다..!

 

 

그래서 for문을 통해 하나씩 곱한 값을 리스트로 저장하고,

그 리스트의 값들을 다시 더하는 구조로 바꿨습니다!

def solution(a, b):
    answer = 0
    dot = [i*j for i,j in zip(a,b)]
    for a in dot :
        answer += a
        
    return answer

 

따로 리스트로 저장하지 않고,

곱하면서 더하면 더 짧아지게 되네용!

def solution(a, b):
    answer = 0
    for i,j in zip(a,b) :
        answer += i*j
    return answer

 

감사합니다 :)

댓글