Numba nopython 模式布尔索引

问：

我正在使用 numba 中的数组，这些数组有时会返回 inf 或 nan 作为条目，然后我需要将其调整为零。我在 nopython 模式下工作，所以我想知道是否有办法使用布尔索引（或乘法）将数组中的任何 nan 或 inf 元素调整为零（数组形状为 （vertex_number， 1））。

显然，我可以像这样循环数组：

from numpy import isnan, isinf, array, zeros, nan, inf, ones

from numba import jit

v = ones((10, 1))

v[0] = nan
v[-1] = inf
print(v)

@jit(nopython=True, nogil=True)
def reciprocal_v(v):
    u = 1 / v
    for i in range(len(u)): # get rid of zero length reciprocals...
        if (isinf(u[i])[0] == True) | (isnan(u[i])[0] == True):
            u[i] = 0.0
    return u
u = reciprocal_v(v)
print(u)

我尝试了布尔乘法，但 inf*0 返回 nan，nan*0 返回 nan

@jit(nopython=True)
def numba_isnan(x): return isnan(x)

@jit(nopython=True)
def numba_isinf(x): return isinf(x)

@jit(nopython=True, nogil=True)
def reciprocal_v_bool(v):
    u = 1 / v
    u = u * ~(numba_isnan(u) | numba_isinf(u))
    return u