如何為應該是運算符函數的參數編寫類型提示？例如，看這個函數：import operatorfrom typing import Dict, Tupledef 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 ...>.編輯：我意識到運算符是一個模塊，而不是類型，并且我正在尋找的參數是該模塊中的函數。我想看看是否有任何方法可以讓用戶知道我期望一個運算符函數作為參數。是在文檔字符串中解釋的唯一方法嗎？
1 回答
慕神8447489
TA貢獻1780條經驗 獲得超1個贊
operator是一個模塊，而不是一個類或類型。您想要做的是創建一個可以滿足傳遞給的函數的類型提示func。如果我們從頭開始編寫operator.lt，它可能看起來像這樣：
from numbers import Number
def le(a: Number, b: Number) -> bool:
...
使用Callable類型我們可以將其寫為：
Callable[[Number, Number], bool]
總而言之，你會得到這樣的結果：
import operator
from typing import Dict, Tuple, Callable, Optional
from numbers import Number
Operator = Callable[[Number, Number], bool]
OperatorDict = Dict[str, Tuple[Operator, float]]
def func(thresholds: Optional[OperatorDict] = None) -> None:
if thresholds is None:
thresholds = {"height": (operator.lt, 0.7), "width": (operator.gt, 0.1)}
pass
添加回答
舉報
0/150
提交
取消