python - 如何在类型提示中指定 pandas 系列元素的类型？

Question

我的函数返回一个 Pandas 系列，其中所有元素都具有特定类型(例如 str )。下面的 MWE 应该会给人一个印象:

import pandas as pd 
def f() -> pd.Series:
    return pd.Series(['a', 'b']) 

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

def f() -> pd.Series[str]:

但

TypeError: 'type' object is not subscriptable

那么，如何在类型提示中指定 pandas 系列元素的类型呢？有任何想法吗？

Answer 1

对于 python 3.8 尝试:

def f() -> "pd.Series[str]":
    pass

或者:

f_return_type = "pd.Series[str]"
def f() -> f_return_type:
    pass

或 # type: pd.Series[str]对于变量