3 回答
TA贡献1869条经验 获得超4个赞
你可以这样做:
arr = numpy.nan_to_num(numpy.log(matrix))
nan_to_num 的行为将所有 NaN 替换为零。
您可以在这里找到更多信息:
另一种选择是将掩码传递给where=
np.log 函数的参数。
TA贡献1789条经验 获得超10个赞
np.log是一个ufunc带where参数的。这告诉它哪些元素x将用于计算。其余的被跳过。这最好与out参数一起使用,如下所示:
In [25]: x = np.array([1.,2,0,3,10,0])
In [26]: res = np.zeros_like(x)
In [27]: idx = x>0
In [28]: np.log(x)
/usr/local/bin/ipython3:1: RuntimeWarning: divide by zero encountered in log
#!/usr/bin/python3
Out[28]:
array([0. , 0.69314718, -inf, 1.09861229, 2.30258509,
-inf])
In [29]: np.log(x, out=res, where=idx)
Out[29]:
array([0. , 0.69314718, 0. , 1.09861229, 2.30258509,
0. ])
添加回答
举报