프로그래밍/Python

프로그래머스_python lv0. 짝수는 싫어요

O'bin 2023. 1. 8. 13:48

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 

 

내 답안 : 

 

1
2
3
def solution(n):
    answer = [i for i in range(n+1if i % 2 == 1]
    return answer
cs

 

range를 이용한 for문과 if문을 사용해서 리스트 만드는 방법을 터득했다.

 

 

 

개선안 : 

 

1
2
def solution(n):
    return [i for i in range(1, n+12)]
cs

 

내가 작성한 if 조건으로 홀수만 뽑았지만,

range의 첫번째 인자 start를 1로 하고, 세번째 인자 step을 2로 지정하면 마찬가지로 홀수만 뽑을 수 있다.

이 방법도 알아둬야겠다.