PEP 484–Type Hints

1
2
3
4

def greeting(name: str) -> str:
	return 'Hello ' + name`
# `name: str`: `name` 인자는 문자열이어야 함을 나타냅니다.
# `-> str`: 함수가 문자열을 반환함을 나타냅니다.

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

# 1. 기본적인 타입 힌트 사용
def greeting(name: str) -> str:
    # name 파라미터는 문자열(str) 타입이어야 함을 나타냄
    # -> str 은 함수가 문자열을 반환함을 나타냄
    return f"Hello, {name}!"

# 2. 여러 기본 타입들의 사용
def calculate_total(quantity: int, price: float) -> float:
    # quantity는 정수(int), price는 실수(float) 타입
    # 반환값은 실수(float) 타입
    return quantity * price

# 3. 리스트 타입 힌트 사용
from typing import List

def get_first_name(names: List[str]) -> str:
    # names는 문자열 리스트임을 나타냄
    # List[str]은 모든 요소가 문자열인 리스트를 의미
    return names[0] if names else ""

# 4. 옵셔널 타입 사용
from typing import Optional

def find_user(user_id: Optional[int]) -> Optional[str]:
    # user_id는 정수이거나 None일 수 있음을 나타냄
    # 반환값도 문자열이거나 None일 수 있음
    if user_id is None:
        return None
    return f"User_{user_id}"

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

from typing import Dict, Tuple, Union, Callable

# 5. 딕셔너리와 복합 타입
def process_user_data(
    user_info: Dict[str, Union[str, int]]
) -> Tuple[str, int]:
    # Dict[str, Union[str, int]]는 
    # - 키는 문자열이고
    # - 값은 문자열 또는 정수인 딕셔너리를 의미
    # Tuple[str, int]는 문자열과 정수로 구성된 튜플을 반환한다는 의미
    name = user_info.get("name", "")
    age = user_info.get("age", 0)
    return name, age

# 6. 함수 타입 힌트
def apply_operation(
    func: Callable[[int, int], int],
    x: int,
    y: int
) -> int:
    # Callable[[int, int], int]는
    # - 두 개의 정수를 입력받고
    # - 정수를 반환하는 함수를 의미
    return func(x, y)

# 7. 제네릭 타입
from typing import TypeVar, Sequence

T = TypeVar('T')  # 제네릭 타입 변수 정의

def first_element(sequence: Sequence[T]) -> Optional[T]:
    # Sequence[T]는 임의의 타입 T로 이루어진 시퀀스를 의미
    # Optional[T]는 T 타입 또는 None을 반환할 수 있음을 의미
    return sequence[0] if sequence else None