如何输入提示传递给函数的 pandas 系列的内容？

Question

我有两个接受

pd.Series

参数的函数

def function_1(ages:pd.Series)->float:
    ...

和

def function_2(names:pd.Series)->int:
    ...

但是，对于

function_1

，输入是整数

pd.Series

，对于

function_2

，输入是字符串

pd.Series

。

我希望能够写：

def function_1(ages:pd.Series[int])->float:
    ...

def function_2(names:pd.Series[str])->int:
    ...

但是这会给出错误：

TypeError: 'type' object is not subscriptable

注意：现在我正在做

def function_1(ages:'pd.Series[int]')->float:
    ...

def function_2(names:'pd.Series[str]')->int:
    ...

Answer 1

您的示例应该在 Python >= 3.9 中运行。

让它在 Python 3.8（可能还有更早的 3.x）中工作

from __future__ import annotations

围绕打字的 PEP 不断发展，Python 也是如此......