我正在使用numba中的数组,这些数组有时会返回inf或nan作为条目,然后我需要将其调整为零。我在nopython模式下工作,所以我想知道是否有一种方法可以使用布尔索引(或乘法)将数组中的nan或inf元素调整为零(数组形状为(vertex_number,1))。显然,我可以像这样循环遍历数组:
from numba import jit,<em
我想做的是如果一个numberotherwise 0->它返回np.nan如果b == 0->它返回一个特定的==,它执行一些非常简单的代数我已经给出了下面的玩具例子(我的实际公式更复杂如果输入是数组,如果输入是标量,则numba是完美的,如果输入是大小为1的数组,numba不工作(Cannot unify array(int64, 0d, C) and float64