python - 过滤 Pandas 列，这是两个日期之间的差异

Question

我有一个显示 2 个日期和它们之间的持续时间的 Pandas 表:

|   date1    |   date2    | duration |
+------------+------------+----------+
| 10/04/2018 | 15/05/2018 |   5 days |
| 23/04/2018 | 28/04/2018 |   5 days |
| 27/11/2018 | 28/11/2018 |   1 days |
+------------+------------+----------+

持续时间列由 df['duration'] = df['date2'] - df['date1'] 生成
如果我尝试过滤低于 5 的持续时间，例如: df.query('duration < 5')这失败了，因为它试图过滤 int 上的日期时间对象。
解决这个问题的最佳方法是什么？将 5 设置为日期时间对象？

Answer 1

您可以使用 .dt.days访问器在 timdelta values 以整数值获取天数，然后您可以将其与另一个数字进行比较。

>>> df.query('duration.dt.days < 5')
       date1      date2 duration
2 2018-11-27 2018-11-28   1 days