在Python中，我如何创建一个类型提示来表示Pandas DataFrames，其中至少有一列包含datetime.date对象？

carvr3hs 于 2023-08-01 发布在 Python

关注(0)|答案(1)|浏览(104)

我有几个在时间序列中运行的函数，比如Pandas DataFrames。这些DataFrames有一个包含datetime.date对象的列。我想使用类型命中来明确地告诉用户，我希望列具有datetime.date类型。
目前，我使用的类型提示如下：

import pandas as pd

def my_function(df: pd.DataFrame) -> pd.Dataframe:
    # [...]
    return new_df

字符串
但这些类型提示并没有太大帮助。

1条答案

可以使用assert语句检查列是否满足各自的数据类型

import pandas as pd

def my_function(df: pd.DataFrame) -> pd.Dataframe:
    assert df['col1'].dtype==date
    return new_df

字符串