Python_음양 더하기

<문제>

어떤 정수들이 있습니다. 이 정수들의 절댓값을 차례대로 담은 정수 배열 absolutes와 이 정수들의 부호를 차례대로 담은 불리언 배열 signs가 매개변수로 주어집니다. 실제 정수들의 합을 구하여 return 하도록 solution 함수를 완성해주세요.

[제한사항]

absolutes의 길이는 1 이상 1,000 이하입니다.
absolutes의 모든 수는 각각 1 이상 1,000 이하입니다.
signs의 길이는 absolutes의 길이와 같습니다.
signs[i] 가 참이면 absolutes[i] 의 실제 정수가 양수임을, 그렇지 않으면 음수임을 의미합니다.

<풀이>

def solution(absolutes, signs):
    arr = []
    for i in range(len(absolutes)):
        if signs[i] == True:
            arr.append(absolutes[i])
        elif signs[i] == False:
            arr.append(-absolutes[i])
    return sum(arr)

<다른 풀이>

def solution(absolutes, signs):
    return sum(absolutes if sign else -absolutes for absolutes, sign in zip(absolutes, signs))

<느낀점>

zip 함수를 통해 absolutes와 signs를 묶고, list comprehension을 이용하여 한줄로 표현한 것이 너무 인상깊었다. 아직까지는 zip함수가 익숙하지가 않아서 표현에 어려움이 있지만 편리한 함수인만큼 다양하게 사용하는 것이 좋을 듯 하다.