python - 如何检查熊猫 DateTimeIndex 日期是否属于列表？

Question

我有一个idx带有分钟频率的熊猫日期时间索引，我也有一个日期列表（或集合）list_of_dates。我想返回一个相同大小的布尔数组，idx条件是日期时间索引所属的日期在list_of_dates. 是否有可能以矢量化方式（即不使用 for 循环）来实现？

标签： pythonpandasdatetimeindex

Answer 1

由于您尝试仅比较日期，因此您可以删除时间并像这样进行比较：

>>> df.index.normalize().isin(list_of_dates)

或者：

>>> df.index.floor('D').isin(list_of_dates)