Numpy 如何在Numba中用多维数组中的数字替换NAN？

在普通python中，用numpy数组中的数字替换nan非常简单。但是，在Numba中执行相同操作时，以下操作失败

@jit(nopython=True)
def dostuff():
    x = np.array([[1,np.nan,3]]);
    np.nan_to_num(x,copy=False);
dostuff()

在numpy数组中，在Numba可编译函数中，如何用零替换NAN？对于一维，我们可以做

x[np.isnan（x）]=0

，但对于更高的维度，这同样失败

对于一维，我们可以做x[np.isnan（x）]=0，但对于更高的 这也失败了

有线索：）

输出：

[[1. 0. 3.]
 [0. 2. 6.]]

对于一维，我们可以做x[np.isnan（x）]=0，但对于更高的 这也失败了

有线索：）

输出：

[[1. 0. 3.]
 [0. 2. 6.]]

---

我想你的意思是

.ravel（）

而不是

.flatte（）

，正如问题所要求的那样，但确实应该这样做，谢谢！是的，这一点很好-我已经相应地修改了代码。我想你的意思是

.ravel（）

，而不是问题要求的

.flatte（）

，但确实应该这样做，谢谢！是的，很好-我已经相应地修改了代码。