且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

在Python中使用numpy/scipy忽略数组中的-Inf值

更新时间:2022-03-20 01:19:50

使用掩码数组 :

>>> a = numpy.array([2, 0, 1.5, -3])
>>> b = numpy.ma.log(a)
>>> b
masked_array(data = [0.69314718056 -- 0.405465108108 --],
             mask = [False  True False  True],
       fill_value = 1e+20)

>>> b.sum()
1.0986122886681096