如何解决Union[None,int]类型变量传入int参数函数的类型错误

当你明确a()一定会返回整数，但类型标注为Union[None, int]时，可通过以下几种方式解决调用b(a())的类型错误：

方法1：显式判空检查（最安全推荐） #

先判断返回值不是None，再传入函数b，类型检查工具会自动识别合法的类型：

from typing import Union

def a() -> Union[None, int]:
    return 10  # 实际确定返回int

def b(i: int):
    pass

result = a()
if result is not None:
    b(result)

方法2：使用类型断言（typing.cast） #

通过cast明确告知类型检查工具，你确定当前返回值是int类型：

from typing import Union, cast

def a() -> Union[None, int]:
    return 10

def b(i: int):
    pass

b(cast(int, a()))

方法3：Walrus运算符+判空（Python 3.8+） #

用海象运算符简化代码，同时完成判空和赋值操作：

from typing import Union

def a() -> Union[None, int]:
    return 10

def b(i: int):
    pass

if (result := a()) is not None:
    b(result)

方法4：临时忽略类型检查（不推荐） #

如果只是临时应急，可通过注释让类型检查工具跳过此行，但这会掩盖潜在的None值风险，不建议长期使用：

from typing import Union

def a() -> Union[None, int]:
    return 10

def b(i: int):
    pass

b(a())  # type: ignore

内容的提问来源于stack exchange，提问作者طاها سمر