음양 더하기 - 리스트를 합쳐서 사용하고 싶을 때 zip

 

해결일	언어	레벨	테스트명
20240618	Python	level 1	음양 더하기

 

링크 : https://school.programmers.co.kr/learn/courses/30/lessons/76501

 

 

 

---

 

내 코드

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

 

 

 

개선점:

거의 똑같은 코드를 두번 썼음

그리고 굳이 sum 하기 전에 리스트화를 하지 않아도 되겠다

 

베스트 코드

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

 

 

 

개수가 같은 두개의 리스트의 값을 둘 다 사용하여 값을 내야 할 때

-> zip 함수 사용하기!!!