在numba函數中使用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) 

但是，您不能創建它出現一個DateTime對象：

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

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