파이썬 or 사용방법

문제 보기

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

프로그래머스

 

 

문제 해결

- 처음 해결한 풀이

def solution(arr, divisor):
    lst = sorted(i for i in arr if i % divisor == 0)
    return lst if len(lst) > 0 else [-1]

파이썬에서는 or 연산자를 사용하여 or 앞이 참일경우 해당 값까지 출력하고, 거짓일경우 뒤를 출력한다.

 

 

소스 코드 (개선 풀이)

def solution(arr, divisor):
    return sorted(i for i in arr if i % divisor == 0) or [-1]