如何在Numba JitClass中转换numpy.datetime64为Unix时间戳

希望直接向Numba JitClass传入numpy.datetime64对象，并在类内部将其转换为浮点型Unix时间戳，而非在创建对象前提前计算时间戳。最终目标是传入起止日期后，转换为Unix时间戳并使用np.arange生成时间戳数组。当前只能在外部提前计算时间戳：

import numpy as np
# 外部计算Unix时间戳示例
(np.datetime64('2024-01-01T00:00:00') - np.datetime64('1970-01-01T00:00:00')) / np.timedelta64(1, 's')
# 输出：1704067200.0

现有JitClass定义如下，需修改以实现内部转换逻辑：

import numpy as np
from numba.experimental import jitclass
from numba import types

spec=[
    ('start', types.NPDatetime('s'))
]

@jitclass(spec)
class Foo():
    def __init__(self, start):
        self.start = start

obj = Foo(np.datetime64('2024-01-01T00:00:00'))

在JitClass内部可通过计算目标时间与Unix纪元时间的差值，转换为浮点型时间戳，具体实现如下：

import numpy as np
from numba.experimental import jitclass
from numba import types

# 定义类成员变量的静态类型，新增存储Unix时间戳的字段
spec = [
    ('start_dt', types.NPDatetime('s')),  # 原始datetime64对象（秒级精度）
    ('start_ts', types.float64),          # 转换后的Unix时间戳（浮点型）
    ('end_dt', types.NPDatetime('s')),
    ('end_ts', types.float64)
]

@jitclass(spec)
class Foo():
    def __init__(self, start_dt, end_dt):
        self.start_dt = start_dt
        self.end_dt = end_dt
        
        # 定义Unix纪元时间（1970-01-01 00:00:00，与传入时间精度一致）
        epoch = np.datetime64('1970-01-01T00:00:00', 's')
        # 计算并存储Unix时间戳
        self.start_ts = (self.start_dt - epoch) / np.timedelta64(1, 's')
        self.end_ts = (self.end_dt - epoch) / np.timedelta64(1, 's')
    
    def generate_timestamp_array(self, step=1):
        # 生成指定步长的时间戳数组
        return np.arange(self.start_ts, self.end_ts + step, step, dtype=np.float64)

# 使用示例
obj = Foo(
    np.datetime64('2024-01-01T00:00:00'),
    np.datetime64('2024-01-01T00:00:05')
)

# 查看转换结果
print(obj.start_ts)  # 输出：1704067200.0
print(obj.end_ts)    # 输出：1704067205.0

# 生成时间戳数组
print(obj.generate_timestamp_array())
# 输出：[1704067200. 1704067201. 1704067202. 1704067203. 1704067204. 1704067205.]

关键注意事项 #

• 精度统一：确保纪元时间、传入的datetime64对象、timedelta64的精度完全匹配（示例中统一使用秒级s），否则会出现转换错误
• 静态类型声明：必须在spec中明确所有成员变量的类型，Numba依赖静态类型信息完成JitClass编译
• 浮点型输出：通过除以timedelta64(1, 's')得到浮点型时间戳，符合多数场景下的使用需求

内容的提问来源于stack exchange，提问作者brokkoo