이펙티브 파이썬 4장 - 컴프리헨션과 제너레이터

Better way 27 map과 filter 대신 컴프리헨션을 사용하라

# map과 filter 사용
numbers = [1, 2, 3, 4, 5, 6]
even_numbers = list(filter(lambda x: x % 2 == 0, numbers))
even_squares = list(map(lambda x: x ** 2, even_numbers))

# 좋은 예시 - 리스트 컴프리헨션
even_squares = [x ** 2 for x in numbers if x % 2 == 0]

 

Better way 28 컴프리헨션 내부에 제어 하위 식을 세 개 이상 사용하지 말라

# 안 좋은 예시
matrix = [[[1, 2], [3, 4]], [[5, 6], [7, 8]]]
flattened = [num for layer in matrix for row in layer for num in row if num % 2 == 0]

# 좋은 예시
matrix = [[[1, 2], [3, 4]], [[5, 6], [7, 8]]]
even_numbers = []
for layer in matrix:
    for row in layer:
        for num in row:
            if num % 2 == 0:
                even_numbers.append(num)

 

Better way 29 대입식을 사용해 컴프리헨션 안에서 반복 작업을 피하라

# 안 좋은 예시
words = ["hello", "world", "python"]
lengths = [len(word) for word in words if len(word) > 4]

# 좋은 예시
lengths = [length for word in words if (length := len(word)) > 4]

Better way 30 리스트를 반환하기보다는 제너레이터를 사용하라

# 리스트 반환
def squares(n):
    return [x ** 2 for x in range(n)]
    
# 제너레이터 사용
def squares(n):
    for x in range(n):
        yield x ** 2

Better way 31 인자에 대해 이터레이션할 때는 방어적이 돼라

# 안 좋은 예시
def process(iterator):
    first_pass = list(iterator)
    second_pass = list(iterator)  # 빈 리스트 반환
    
# 이터레이터를 리스트로 변환
def process(iterable):
    data = list(iterable)
    first_pass = data
    second_pass = data

 

Better way 32 긴 리스트 컴프리헨션보다는 제너레이터 식을 사용하라

# 리스트 컴프리헨션 (많은 메모리 차지)
squares = [x ** 2 for x in range(10**6)]

# 제너레이터 ( next() 를 호출 할 때 마다 호출)
squares = (x ** 2 for x in range(10**6))

Better way 33 yield from을 사용해 여러 제너레이터를 합성하라

# 안좋은 예시 반복문 사용
def chain(iter1, iter2):
    for item in iter1:
        yield item
    for item in iter2:
        yield item

# 좋은 예시 - yield from 사용
def chain(iter1, iter2):
    yield from iter1
    yield from iter2

 

Better way 34 send로 제너레이터에 데이터를 주입하지 말라

# 안 좋은 예시 외부에서 send()로  값을 받아옴
def message_generator():
    message = "Start"
    while True:
        received = yield message  # 외부에서 값을 받아옴
        if received is not None:
            message = f"Received: {received}"

msg_gen = message_generator()
print(next(msg_gen))  # Start
print(msg_gen.send("Hello"))  # Received: Hello

# 더 나은 예시 - 함수 인자로 데이터 전달
def process_message(message, additional):
    return f"Processed: {message} with {additional}"

print(process_message("Hello", "Extra Data"))  # Processed: Hello with Extra Data

Better way 35 제너레이터 안에서 throw로 상태를 변화시키지 말라

# 안 좋은 예시 - throw()를 사용
def error_generator():
    try:
        while True:
            yield
    except ValueError:
        print("Error received!")

gen = error_generator()
next(gen)
gen.throw(ValueError)  # Error received!

# 좋은 예시
def safe_generator():
    yield "Processing..."
    raise ValueError("Error occurred")

try:
    gen = safe_generator()
    print(next(gen))
    print(next(gen))  # 여기서 예외 발생
except ValueError as e:
    print("Handled error:", e)

Better way 36 이터레이터나 제너레이터를 다룰 때는 itertools를 사용하라

# 안 좋은 예시 - 수동 반복 구현
def repeat_n_times(value, n):
    return [value for _ in range(n)]

print(repeat_n_times("Hello", 3))  # ['Hello', 'Hello', 'Hello']


# 좋은 예시 - itertools 사용
from itertools import repeat
print(list(repeat("Hello", 3)))  # ['Hello', 'Hello', 'Hello']