파이썬 - list comprehension

코딩테스트 연습 - 두 정수 사이의 합 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

 

def solution(a, b):
    if a > b :
        k = [x for x in range(b,a+1)]
    elif a==b : 
        k = [a]
    elif a<b : 
        k = [x for x in range(a,b+1)]
    return sum(k)

 

comprehension = 포함, 함축, 압축, 이해

 

x for x 라는 내장 함수를 생각없이 쓰고 있었는데 이게 list comprehension 이다.

이에 더해 알아 둬야 할 것은 이것을 정의 할 때엔 앞에서

즉, k = 으로 정의해야 작동하지 [] = k 로 하면 오류가 발생한다.

[] = k가 아닌  []==k , T/F판문만 작동하게 된다. 

 

+a 로 range 는 (x,y)범위 내라고 생각하는 것이 편하다.