celina의 이것저것

List Comprehension 본문

자료구조&알고리즘/프로그래머스

List Comprehension

celinayk 2024. 4. 29. 21:19
반응형
def solution(n):
    ans = []
    for i in range(1, n+1, 2):
        ans.append(i)
    return ans

원래는 이런 코드인데

 

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

이렇게 한줄컷이 가능하다;;; 대박

 

=> 리스트 컴프리헨션이라고 부르는 파이썬 문법이다

[ ( 변수를 활용한 값 ) for ( 사용할 변수 이름 ) in ( 순회할 수 있는 값 )]

이렇게 쓴다고 한다 

 

size = 10
arr = [i * 2 for i in range(size)]

print(arr)

[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

이런느낌?

개쩐다 이래서 다들 파이썬으로 갈아타는군

근데 파이썬 문법 어색해서 아직 원래 문법이 편하다 ㅋ

Comments