python - 检查 Pandas DataFrame 索引是否为特定类型的正确方法 (DatetimeIndex)

Question

在下面的代码中，我想检查 index数据帧中的类型为 DatetimeIndex .这是这样做的正确方法吗？有没有比 if 更好的方法来做到这一点？陈述？看起来很简单，但想知道我是否遗漏了什么。

我想这样做是因为我预见到用户有时可能不会阅读 data与相应的日期信息。

import pandas as pd
data = {'x' : [1,2,3],
    'y' : [4,5,6]}

index = pd.date_range("2014-1-1", periods=3, freq="D")

情况1

df = pd.DataFrame(data)
type(df.index) == pd.tseries.index.DatetimeIndex

正确返回: False
案例二

df = pd.DataFrame(data, index=index)
type(df.index) == pd.tseries.index.DatetimeIndex

正确返回: True
提前致谢。我希望这不是太琐碎。

Answer 1

截至 2021 年，以下是检查这一点的最新方法:

>>> df.index.inferred_type == "datetime64"
>>> True

所以你可以在你的应用程序中添加这样的东西:

assert df.index.inferred_type == 'datetime64', "must have a datetime index"

干杯