파이썬 코드카타

내적 # zip()

여연찌 2024. 4. 11. 14:46
  • 길이가 같은 두 1차원 정수 배열 a, b가 매개변수로 주어집니다. a와 b의 내적을 return 하도록 solution 함수를 완성해주세요.
  • 이때, a와 b의 내적은 a[0]*b[0] + a[1]*b[1] + ... + a[n-1]*b[n-1] 입니다. (n은 a, b의 길이)
def solution(a, b):
    answer = [x*y for x,y in zip(a,b)]
    result = sum(answer)
    return result

① answer = [x*y for x,y in zip(a,b)]

# zip() 함수를 사용하여 a와 b 리스트를 엮어준다

# 엮은 리스트에서 x 와 y로 임의의 값을 주고 x*y 를 계산해준다

# 계산한 값은 리스트 인수로 반환

 

② result = sum(answer)

# 반환된 리스트 값을 합해준다

* return sum(answer)

이렇게 더해줘도 됨

 

 


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

① for i in range(len(a)):

# 0부터 a 의 숫자 만큼 반복해준다

 

②  answer += a[i]*b[i]

# a 와 b 의 i 숫자만큼의 인덱스를 곱하여 answer 에 더해준다

 

 

 

 

 

zip() 함수 기본 문법

for x,y in zip(a,b)

 

 

zip() 함수로 x*y 를 계산해준 값

answer = 0
for x,y in zip(a,b):
    answer += x*y