Python如何用一行代码强制Tuple返回值符合指定固定长度类型要求

问题原因 #

类型检查器无法自动识别生成器表达式/split()方法返回结果的固定长度，默认会将动态构造的元组判定为不定长的Tuple[str, ...]类型，和你声明的固定长度二元元组返回类型不匹配，因此报错。

可行的一行解决方案 #

这里提供两种生产环境常用的实现，都可以满足要求：

• 显式解包构造（推荐，自带运行时校验，输入不符合双元素格式时直接抛出错误，无需额外导入）

def function(input: str) -> Tuple[str, str]:
    a, b = input.split(","); return (a, b)

如果需要避免输入存在多余逗号时拆分异常，可以加maxsplit参数优化：

def function(input: str) -> Tuple[str, str]:
    return (*input.split(",", maxsplit=1),)[:2]

• 类型强制转换（仅类型检查层面生效，无运行时开销，适合确定输入格式绝对符合要求的场景）

from typing import Tuple, cast

def function(input: str) -> Tuple[str, str]:
    return cast(Tuple[str, str], tuple(input.split(",")))

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