22. 변수 위치 인자를 사용해 시각적인 잡음을 줄여라¶

위치 인자 (Positional argument) 를 가변적으로 받을 수 있으면 함수 호출이 더 깔끔해지고 시각적 잡음도 줄어든다.

In [4]:

def log(message, values):
    if not values:
        print(message)
    else:
        values_str = ', '.join(str(x) for x in values)
        print(f'{message}: {values_str}')

In [5]:

log('내 숫자는', [1, 2])

내 숫자는: 1, 2

In [6]:

log('안녕', [])

안녕

빈 리스트를 넘겨야 한다면 귀찮음 뿐만 아니라 잡음이 많다.

생략 가능하도록 하자

In [7]:

def log(message, *values):
    if not values:
        print(message)
    else:
        values_str = ', '.join(str(x) for x in values)
        print(f'{message}: {values_str}')

In [10]:

log('내 숫자는', 1, 2)

내 숫자는: 1, 2

In [11]:

log('안녕')

안녕

언패킹 대입문에 쓰인 별표식과 비슷함

시퀀스를 사용하고 싶다면 * 연산자를 사용하면 된다.

In [12]:

favorite = [1, 2, 3]
log('내 숫자는', *favorite)

내 숫자는: 1, 2, 3

In [13]:

favorite = [1, 2, 3]
log('내 숫자는', favorite)

내 숫자는: [1, 2, 3]

가변적인 위치 인자를 받는 데는 두 가지 문제점이 있다.

첫번째 : 선택적인 위치 인자가 함수에 전달되기 전에 항상 튜플로 변환됨

메모리를 아주 많이 소비하거나 프로그램이 중단될 수 있음.

In [21]:

def my_generator():
    for i in range(10):
        yield i

In [22]:

def my_func(*args):
    print(args)

In [23]:

it = my_generator()

In [24]:

my_func(*it)

(0, 1, 2, 3, 4, 5, 6, 7, 8, 9)

두 번째 문제점은 새로운 위치 인자를 추가하면 해당 함수를 호출하는 모든 코드를 변경해야 한다.

In [25]:

def log(sequence, message, *values):
    if not values:
        print(f'{sequence} - {message}')
    else:
        values_str = ', '.join(str(x) for x in values)
        print(f'{sequence} - {message}: {values_str}')

In [26]:

log(1, '좋아하는 숫자는', 7, 33)

1 - 좋아하는 숫자는: 7, 33

In [27]:

log(1, '안녕')

1 - 안녕

In [28]:

log('좋아하는 숫자는', 7, 33)

좋아하는 숫자는 - 7: 33

이런 가능성을 완전히 없애려면 *args를 받아들이는 함수를 확장할 떄는 키워드 기반의 인자만 사용해야 한다. (Better way 25)

더 방어적으로 프로그래밍하려면 타입 애너테이션 (Better way 90)을 사용해도 된다.

기억해야 할 내용¶

def 문에서 *args를 사용하면 함수가 가변 위치 인자를 받을 수 있다.
* 연산자를 사용하면 가변 인자를 받는 함수에게 시퀀스 내의 원소들을 전달할 수 있다.
제너레이터에 * 연산자를 사용하면 프로그램이 메모리를 모두 소진하고 중단될 수 있다.
*args를 받는 함수에 새로운 위치 기반 인자를 넣으면 감지하기 힘든 버그가 생길 수 있다.