如何在类型提示中指定系列元素的类型?

My函数返回一个pandas系列,其中所有元素都有一个特定的类型(比如str)。以下MWE应给人留下深刻印象:

将熊猫作为pd导入
def f()->pd.系列:
返回pd.系列(['a','b'])

在我想说明的类型提示中,f()[0]将始终是str类型(例如,与返回pd.Series([0,1])的函数相比)。我这样做:

def();pd.系列[str]:

但是

TypeError:“type”对象不可下标

那么,如何在类型提示中指定系列元素的类型呢?。有什么想法吗

您可以利用typing.TypeVar来完成以下任务:

输入导入的

(
类型变量
)
serieString=TypeVar('pandas.core.series.series(str)'))
def f()->序列字符串:

发表评论