노력에는 지름길이 없으니까요

나누어 떨어지는 숫자 배열 - return에서 or 사용법 본문

Python/프로그래머스

나누어 떨어지는 숫자 배열 - return에서 or 사용법

데건 2024. 6. 24. 10:30
728x90

 

해결일 언어 레벨 테스트명
20240624 Python level 0  나누어 떨어지는 숫자 배열

 

링크 : https://school.programmers.co.kr/learn/courses/30/lessons/12910/solution_groups?language=python3&type=all

 

 

 


 

내 코드

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

 

 

 

개선점: 실행시킬 때만 해도 딱히 없다고 생각했는데 베스트 코드에서 배울 것이 있어서 글을 게시한다.

 

베스트 코드

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

 

or를 쓰면 리스트에 아무것도 없을 때에 저게 리턴되나보다...

728x90
반응형