level1_나누어 떨어지는 숫자 배열

문제 설명

array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요.divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하세요.

 

입출력 예

arr	divisor	return
[5,9,7,10]	5	[5,10]
[2,36,1,3]	1	[1,2,3,36]
[3,2,6]	10	[-1]

 

내가 제출한 코드

def solution(arr, divisor):
    answer = []
    for i in arr:
        if i % divisor == 0:
            answer.append(i)
    
    if not answer:  # answer 리스트가 비어 있으면 -1을 반환
        return [-1]
    
    answer.sort()
    return answer.sort()

 

다른 사람이 제출한 코드

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

 

우선 (n % divisor == 0). 나누어 떨어지는 요소들만 새로운 리스트에 포함시킨다.

위에서 생성된 리스트는 sorted 함수에 의해 오름차순으로 정렬한다.

만약 arr에서 divisor로 나누어 떨어지는 요소가 없다면, 리스트 컴프리헨션의 결과는 빈 리스트가 된다.

파이썬에서 빈 리스트는 불리언 'False'로 간주된다. 따라서 다. 따라서 sorted()의 결과가 빈 리스트일 경우, or 연산자의 오른쪽 피연산자인 [-1]이 결과값으로 사용된다.

 

한줄로.... 이렇게 사용할 수 있다니... 다시 배워간다.. 파이썬