Python의 함수

Python의 함수에 대한 노트

Ellipsis¶

Ellipsis는 말줄임표를 말하는데, Python에서 Ellipsis라고 할 경우 ...을 말한다. 함수나 제어문에서 pass를 대신하여 사용할 수 있다.

for i in range(10):
    if i == 5:
        ...

    print(i, end='')

0123456789

def test(): ...

print(test())

None

parameter의 default값 지정¶

parameter의 default값은 아래와 같이 지정할 수 있다. __defaults__ 속성을 사용하면 각 parameter에 지정된 default값을 확인할 수 있다.

def add(a=1, b=False):
    return a + b

print(add.__defaults__)

(1, False)

print(add(1))

Tip

함수에 입력값이 없어도 되는 parameter를 만들고 싶을 때는 default 값을 상황에 따라 False또는 None으로 지정하면 된다.

Info

default값이 설정된 parameter는 default값이 설정되지 않는 parameter보다 반드시 뒤에 나와야 한다.

Function Annotation¶

Python의 함수를 작성할 때, 다음과 같이 각 parameter와 return의 형태에 대한 주석을 달아서 작성할 수 있다. #을 이용해서 표기한 주석과 마찬가지로 어떠한 강제성도 없다. __annotations__ 속성을 사용하면 내용을 확인할 수 있다.

def add(a: 'expression', b: int) -> int:
    return a + b

print(add.__annotations__)

{'a': 'expression', 'b': <class 'int'>, 'return': <class 'int'>}

Documentation Strings¶

함수에 대한 설명(Docstring)을 코드 내부에 작성하고 싶을 때는 아래와 같이 작성하면 된다. __doc__ 속성을 사용해서 내용을 확인할 수 있다.

def add(a, b):
    """this is test function"""
    return a + b

print(add.__doc__)

this is test function

*args, **kwargs¶

자료구조의 packing과 unpacking을 이용해서 하나의 파라미터가 여러 개의 인자를 입력 받거나 여러 개의 파라미터가 하나의 해시테이블로 입력되는 함수를 만들 수 있다.

Info

tuple이나 list를 unpacking 할때는 *, dict를 unpacking 할때는 **를 사용한다.

a = [1, 2]
b = [3, 4]
c = [a, b, b, b]

def v_add(*a):  # packing
    res = [sum(v) for v in zip(*a)]  # unpacking
    return res

res = v_add(*c)  # unpacking


print(res)

[10, 14]

def test(**kwargs):  # packing
    return kwargs


data = {"name": "John Doe", "age": 30}
print(test(**data))  # unpacking
# print(test(name="John Doe", age=30))  # 동일한 결과 출력

{'name': 'John Doe', 'age': 30}

Positional-Only/Keyword-Only Arguments¶

함수나 클래스를 선언할 때 Positional Arguments 또는 Keyword Arguments만 입력 받도록 할 수 있다.

Keyword-Only Arguments¶

아래와 같이 *를 사용해서 함수나 클래스의 파라미터가 인자를 반드시 kwargs 방식으로만 입력 받을 수 있도록 강제할 수 있다.

def func(*, a):
    return a


func(1)

Traceback (most recent call last):
  File "C:\projects\python311\main.py", line 5, in <module>
    func(1)
TypeError: func() takes 0 positional arguments but 1 was given

class MyClass:
    def __init__(self, *, a) -> None:
        self.a = a


mc = MyClass(1)

Traceback (most recent call last):
  File "C:\projects\python311\main.py", line 6, in <module>
    mc = MyClass(1)
         ^^^^^^^^^^
TypeError: MyClass.__init__() takes 1 positional argument but 2 were given

Positional-Only Arguments¶

개인적으로 그다지 선호하지는 않지만 아래와 같이 /를 사용해서 함수나 클래스의 파라미터가 인자를 반드시 args 방식으로만 입력 받을 수 있도록 강제할 수 있다.

def func(a, /):
    return a


func(a=1)

Traceback (most recent call last):
  File "C:\projects\python311\main.py", line 5, in <module>
    func(a=1)
TypeError: func() got some positional-only arguments passed as keyword arguments: 'a'

class MyClass:
    def __init__(self, a, /) -> None:
        self.a = a


mc = MyClass(a=1)

Traceback (most recent call last):
  File "C:\projects\python311\main.py", line 6, in <module>
    mc = MyClass(a=1)
         ^^^^^^^^^^^^
TypeError: MyClass.__init__() got some positional-only arguments passed as keyword arguments: 'a'

함수의 호출¶

아래와 같이 하나의 함수에 여러 이름을 부여해 줄 수 있다.

def func(a):
    return a + 1


expr1 = expr2 = func

a = 3
print(expr1(a))
print(expr2(a))

4
4

아래와 같이 함수를 자료구조에 담아서 호출할 수도 있다.

def func1(a):
    return a + 1


def func2(a):
    return a + 2


a = 3
list_func = [func1, func2]
for f in list_func:
    print(f(a))

4
5

Note

이런 호출이 가능한 이유는 Python이 함수를 일급객체(first-class object)로 취급하는 언어이기 때문에 가능하다.