如何在Python中指定方法返回列表的固定长度类型标注？

针对你的问题，有几种不同层面的解决方案，分别从静态类型检查和运行时强制保证两个角度来实现：

1. 静态类型检查层面 #

如果可以接受返回元组（元组本身是长度固定的不可变结构），直接修改返回类型为tuple[int, int]，mypy、pyright这类类型检查工具会自动识别这是长度为2的整数序列：

def return_list_with_length_2(self) -> tuple[int, int]:
    return (0, 1)

如果必须返回列表，Python标准类型提示无法直接指定列表长度，但可以通过以下方式实现静态校验：

• 借助第三方库pydantic的conlist类型，它支持限制列表的最小/最大长度：

from pydantic import conlist

def return_list_with_length_2(self) -> conlist(int, min_length=2, max_length=2):
    return [0, 1]

配合pydantic的类型校验机制，能在静态检查阶段识别长度不符合的问题。

2. 运行时强制保证层面 #

如果需要在代码运行时确保返回列表长度严格为2，可以用以下两种方式：

方法一：直接添加断言 #

在方法内部对返回结果做长度校验，不符合时直接抛出错误：

def return_list_with_length_2(self) -> list[int]:
    result = [0, 1]
    assert len(result) == 2, "返回的列表长度必须为2"
    return result

方法二：用装饰器复用校验逻辑 #

如果多个方法都需要类似的长度校验，写一个通用装饰器来统一处理：

def ensure_list_length(length):
    def decorator(func):
        def wrapper(*args, **kwargs):
            result = func(*args, **kwargs)
            if isinstance(result, list) and len(result) != length:
                raise ValueError(f"返回列表长度必须为{length}")
            return result
        return wrapper
    return decorator

@ensure_list_length(2)
def return_list_with_length_2(self) -> list[int]:
    return [0, 1]

被装饰的方法返回列表时，会自动校验长度，不符合要求则抛出ValueError。

3. 进阶自定义类型（静态检查） #

可以自定义类型并配置类型检查工具（如mypy）来识别固定长度的列表，比如创建一个List2Int类型：

from typing import List, NewType

List2Int = NewType('List2Int', List[int])

之后需要在mypy的配置中添加自定义规则或借助插件，让它识别List2Int必须是长度为2的列表（此方案相对复杂，适合有定制化类型检查需求的场景）。

内容的提问来源于stack exchange，提问作者coding_noob