[프로그래머스/파이썬] Lv.0 조건에 맞게 수열 변환하기 1 (5')

Programmers 레벨 0_코딩 기초 트레이닝 문제 풀이

https://school.programmers.co.kr/learn/challenges/training?order=recent

코딩 기초 트레이닝 | 프로그래머스 스쿨

 

 

풀이 날짜 : 2023-09-19

소요 시간 : 5min

 

---

 

문제 설명

정수 배열 arr가 주어집니다. arr의 각 원소에 대해 값이 50보다 크거나 같은 짝수라면 2로 나누고, 50보다 작은 홀수라면 2를 곱합니다. 그 결과인 정수 배열을 return 하는 solution 함수를 완성해 주세요.

 

 

 

나의 풀이

def solution(arr):
    for i, a in enumerate(arr) : 
        if a >= 50 and a % 2 == 0 :
            arr[i] = a/2
        elif a <= 50 and a % 2 != 0 :
            arr[i] = a*2
    return arr

 

 

 

다른 풀이

def solution(arr):
    return [num/2 if num>=50 and num%2==0 else (num*2 if num<50 and num%2==1 else num) for num in arr]

컴프리헨션 구문에서 if elif else로 조건을 여러 개 두고 싶을 때는 else () 안에 하나의 if 문을 더 중첩하여 사용하면 된다.