在numba nopython函数中使用numpy.datetime？

Question

是否可以在@jit（nopython = True）函数中创建NPDatetime对象？据我所知，初始化这些对象需要使用字符串对象，这似乎阻止了nopython函数的编译。

Answer 1

你当然可以在numpy日期时间进行操作。

import numpy as np
import numba as nb

@nb.njit
def diff_dt(a, b):
    return a - b

x = np.datetime64('2005-02-25')
y = np.datetime64('2005-02-27')

diff_dt(x, y)

但是，您无法创建它出现的日期时间对象：

@nb.njit
def create_dt(s):
    return np.datetime64(s)

# Fails
create_dt('2005-02-25')

Answer 2

我正面临类似的问题。我所做的只是将日期时间转换为自该纪元以来的整数秒，然后在执行jit操作之后，将其重铸为日期时间。