[Python] 프로그래머스 조건에 맞게 수열 변환하기 1

문제 설명

정수 배열 arr가 주어집니다. arr의 각 원소에 대해 값이 50보다 크거나 같은 짝수라면 2로 나누고, 50보다 작은 홀수라면 2를 곱합니다. 

그 결과인 정수 배열을 return 하는 solution 함수를 완성해 주세요.

 

https://school.programmers.co.kr/learn/courses/30/lessons/181882

프로그래머스

제한 사항

1 ≤ arr의 길이 ≤ 1,000,000
1 ≤ arr의 원소의 값 ≤ 100

예제

arr	result
[1, 2, 3, 100, 99, 98]	[2, 2, 6, 50, 99, 49]

문제 풀이

def solution(arr):
    answer = []
    for i in arr:
        if i >= 50 and i % 2 == 0:
            answer.append(i//2)
        elif i < 50 and i % 2 != 0:
            answer.append(i*2)
        else:
            answer.append(i)
    return answer

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

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.