python 類型提示可從某些模塊/子類調用

Question

我可以使用Callable來輸入參數是 function 本身的提示：

from typing import Callable
def myfun(f: Callable):
    pass

我能否以某種方式指定Callable應該來自，例如pandas.DataFrame.agg （例如sum ）？ 這是（不）推薦的嗎？

Answer 1

如果您想提示 function 用戶使用已知函數的某個子集，我建議直接使用可調用的Enum 。

像這樣的東西：

from enum import Enum
import pandas as pd 


class FunctionAllowed(Enum):

    agg = pd.DataFrame.agg
    sum = pd.DataFrame.sum


def myfun(f:FunctionAllowed):
    a = pd.DataFrame()

    f.value(a)

myfun(FunctionAllowed.sum)