我有一个数组:
x = numpy.array([-inf, -inf, 37.49668579])
有没有办法将-inf值更改为0?
答案 0 :(得分:51)
有:
from numpy import inf
x[x == -inf] = 0
答案 1 :(得分:12)
使用isneginf
http://docs.scipy.org/doc/numpy/reference/generated/numpy.isneginf.html#numpy.isneginf
x[numpy.isneginf(x)] = 0
答案 2 :(得分:0)
使用numpy.nan_to_num
可能更容易,更灵活:
numpy.nan_to_num(x, neginf=0)
Out[1]: array([ 0. , 0. , 37.49668579])