运算符模块的 Python3 类型提示

如何为应该是运算符函数的参数编写类型提示?例如,看这个函数:


import operator

from typing import Dict, Tuple


def func(thresholds: Dict[str, Tuple[operator, float]] = None):

    if thresholds is None:

        thresholds = {"height": (operator.lt, 0.7), "width": (operator.gt, 0.1)}

    pass

当我尝试运行我的代码时,我得到:


TypeError: Tuple[t0, t1, ...]: each t must be a type. Got <module 'operator' from ...>.

编辑:我意识到运算符是一个模块,而不是类型,并且我正在寻找的参数是该模块中的函数。我想看看是否有任何方法可以让用户知道我期望一个运算符函数作为参数。是在文档字符串中解释的唯一方法吗?


汪汪一只猫
浏览 94回答 1
1回答

慕神8447489

operator是一个模块,而不是一个类或类型。您想要做的是创建一个可以满足传递给的函数的类型提示func。如果我们从头开始编写operator.lt,它可能看起来像这样:from numbers import Numberdef le(a: Number, b: Number) -> bool:&nbsp; &nbsp; ...使用Callable类型我们可以将其写为:Callable[[Number, Number], bool]总而言之,你会得到这样的结果:import operatorfrom typing import Dict, Tuple, Callable, Optionalfrom numbers import NumberOperator = Callable[[Number, Number], bool]OperatorDict = Dict[str, Tuple[Operator, float]]def func(thresholds: Optional[OperatorDict] = None) -> None:&nbsp; &nbsp; if thresholds is None:&nbsp; &nbsp; &nbsp; &nbsp; thresholds = {"height": (operator.lt, 0.7), "width": (operator.gt, 0.1)}&nbsp; &nbsp; pass
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python