arrays - 来自外部库的 python numba numpy 数组

Question

我有一个来自 FEniCS 函数对象（例如名为 xval）的数组，通常由np.array(xval.vector()). 使用 numba 如下（除 tbl 之外的其他值是floator int）：

@jit(nopython=True)
def updateMatdataGen(xval,tbl,minTbl,maxTbl,scaler):
    npXval = np.array(xval.vector(),dtype=np.float32)
    xclp = np.clip((npXval-minTbl)/(maxTbl-minTbl)*scaler,0,scaler).astype(int)
    return tbl[xclp]

结果non-precise type pyobject。xval 来自 C++ 库，很容易与 numpy 交换。错误已经在行中npXval = np.array(xval.vector(),dtype=np.float32)。

我能做些什么？

标签： arraysnumpytypesnumba