Numpy与Python中的datetime比较

在本文中，我们将介绍如何使用Numpy比较一个日期时间列表与Python中的datetime对象。

首先，我们需要导入必要的库。

import numpy as np
from datetime import datetime

接下来，我们创建一个包含日期时间的列表以及一个datetime对象。

dt_list = ['2022-01-01 12:00:00', '2022-01-02 12:00:00', '2022-01-03 12:00:00']
dt_obj = datetime(2022, 1, 2, 12, 0, 0)

现在，我们可以使用numpy的datetime64()函数将日期时间列表中的每个元素转换为numpy的datetime64类型。

dt_list_np = np.array(dt_list, dtype='datetime64')

接着，我们可以使用numpy的logical_and()函数将日期时间列表中的每个元素与datetime对象进行比较。

result = np.logical_and(dt_list_np > dt_obj, dt_list_np < np.datetime64('2022-01-04'))

现在，result是一个布尔数组，其中True表示对应的日期时间在datetime对象之后并且在’2022-01-04’之前。我们可以使用numpy的where()函数找到这些日期时间所在的索引。

indices = np.where(result)
print(indices)

输出结果如下：

(array([1]),)

这表示包含’2022-01-02 12:00:00’的日期时间列表中的索引为1。

阅读更多：Numpy 教程

总结

使用numpy比较日期时间与datetime对象可以在处理时间序列数据时非常有用。通过将日期时间转换为numpy的datetime64类型，我们可以使用各种函数比较日期时间。