检查 Pandas 日期时间列是否在一定时间范围内

Question

我有一个带有“日期时间”列的 Pandas 数据框，其中包含一个日期时间:)

ColA    Datetime              ColB
----    -------------------   ----
1       2021-01-01 05:02:22   SomeVal
2       2021-01-01 01:01:22   SomeOtherVal

我想创建一个只有两个规则的新日期列：

1. If the "time" element of datetime is between 00:00:00 and 02:00:00 then make Date the "date" element of Datetime - 1 (the previous day)
2. Otherwise make Date the "date" element of Datetime as is

为了实现这一点，我将不得不对日期时间列进行检查。 那会怎样？ 另外，如果我不需要迭代数据框来实现这一点，我会加分。

Answer 1

将值转换为日期时间，如果时间小于02:00:00在Series.mask减去一天：

from datetime import time

df['Datetime'] = pd.to_datetime(df['Datetime'])

df['Datetime'] = df['Datetime'].mask(df['Datetime'].dt.time <= time(2, 0, 0), 
                                     df['Datetime'] - pd.Timedelta('1 day'))
print (df)
   ColA            Datetime          ColB
0     1 2021-01-01 05:02:22       SomeVal
1     2 2020-12-31 01:01:22  SomeOtherVal