Python - Generators vs list comprehension

이주명·2021년 11월 10일

generator lazy evaluation list comprehension python

목표

List comprehension이란?
Generator는?
List comprehension과 Generator의 차이점
genertator를 사용하는 이유

List comprehension이란?

list comprehension은 list를 쉽고 빠르게 만들수 있는 방법이다.

예를 들어 요소 1, 2, 3, 4, 5 를 가지고있는 리스트를 만들고 싶다고 가정한다.
예시

L = []
for i in range(0,6) :
	L.append(i)

List comprehension을 이용하면 간단하게 한줄로 표기가 가능하다.

L = [i for i in range(0,6)]

my_list_comprehension = [변수, for문, 조건문]

위 형태의 형식을 따르면 생성이 가능하다.

잠깐! 여기서 if 문을 사용할때 break, countinue 같은 if문안에 쓰는 용어는 사용이 불가능하다.

Generator란?

generator는 모든 값을 메모리에 저장하지 않고 그때 그때 값을 생성해서 반환하기 때문에 iterable을 사용할때 한번에 한번의 값만 순환시킨다.

특이한 점은 return 대신 yield를 사용한다는 것이다.

yield는 값을 반환시킨후 함수를 정지시킨다. return 은 값을 반환하고 그대로 함수가 종료되는 면에서 둘은 차이점을 보인다.

L = (i for i in range(0,6))

Generator는 ()를 사용한다.

List comprehension 과 Generator의 차이점

둘의 차이점은

문법

List_comprehension = [z for i in my_list]  
Generator = (z for i in my_list)

[]를 사용하는것과 ()를 사용하는 차이점이 있다.

return
List comprehension은 List를 반환하지만
Generator은 generator를 반환한다.
메모리
List의 경우는 list 안에 속한 모든 데이터를 메모리에 적재하기 때문에 list의 크기 만큼 차지하는 메모리 사이즈가 늘어나게 된다.
하지만 generator는 데이터 값을 한꺼번에 메모리에 적재 하는 것이 아니라 next() 메소드를 통해 차례로 값에 접근할 때마다 메모리에 적재하는 방식이다.
시간

예시로 보겠슴니다

import time

def print_iter(iter):
    for element in iter:
        print(element)

def lazy_return(num):
    print("sleep 1s")
    time.sleep(1)
    return num

print("comprehension_list=")
comprehension_list = [ lazy_return(i) for i in L ]
print_iter(comprehension_list)

print("generator_exp=")
generator_exp = ( lazy_return(i) for i in L )
print_iter(generator_exp)