Python函数如何用类型提示指定返回值类型 Python函数返回值注解的设置技巧​

->

def greet(name: str) -> str:
  """
  这个函数接收一个字符串类型的名字，并返回一个字符串类型的问候语。
  """
  return f"Hello, {name}!"

def add(x: int, y: int) -> int:
  """
  这个函数接收两个整数，并返回它们的和。
  """
  return x + y

def process_data(data: list) -> dict:
  """
  假设这个函数处理一个列表并返回一个字典。
  """
  # 实际的处理逻辑
  result = {}
  for item in data:
      result[item] = len(item)  # 假设字典的键是列表中的项，值是项的长度
  return result

# 甚至可以返回None
def do_something(x: int) -> None:
    """
    这个函数执行一些操作，但不返回任何值。
    """
    print(f"Doing something with {x}")
    # 没有 return 语句，默认返回 None

str

int

list

dict

typing

list

dict

Tuple

Optional

Any

typing.Union

from typing import Union

def get_value(key: str) -> Union[str, int, None]:
    """
    这个函数根据键获取值，可能返回字符串、整数或None。
    """
    data = {"name": "Alice", "age": 30}
    if key == "name":
        return data["name"]
    elif key == "age":
        return data["age"]
    else:
        return None

|

Union

def get_value(key: str) -> str | int | None:
    """
    这个函数根据键获取值，可能返回字符串、整数或None。
    """
    data = {"name": "Alice", "age": 30}
    if key == "name":
        return data["name"]
    elif key == "age":
        return data["age"]
    else:
        return None

mypy

mypy

pip install mypy

mypy

mypy your_file.py

mypy

:param:

:return:

typing.Generator

Generator

None

yield

None

from typing import Generator

def fibonacci(n: int) -> Generator[int, None, None]:
    """
    生成斐波那契数列的前n个数字。
    """
    a, b = 0, 1
    for _ in range(n):
        yield a
        a, b = b, a + b

# 使用示例
for num in fibonacci(10):
    print(num)

Generator[int, None, None]

fibonacci