Python - Index(), OR

카타

 

< 문제26. 음양 더하기  INDEX 활용>

def solution(absolutes, signs):
    temp = []
    for a in absolutes:
      if signs[absolutes.index(a)] == True:
        temp.append(a)
      else:
        temp.append(-a)
    return sum(temp)

index함수를 어떻게 사용하는지 잠깐 헷갈렸어서 기록.

 

---

 

<문제29. 제일 작은 수 제거하기 : OR>

def solution(arr):
    arr.remove(min(arr))
    return arr or -1

array에서 제일 작은 수를 제거하고서, 만약에 array에 남는 값이 없으면 -1을 반환하는 문제.

어제 공부했던 or이 나와서 바로 활용해봤다.

 

*주의

arr.remove(min(arr))을 바로 return 하면 출력값이 None이 된다.

즉, 없애버린 요소를 반환하게  되는것이니 return 값은 arr로 해야함.