假设你有一个像
F = lambda x: sin(x)/x
评估F(0.0)
将导致除以零警告,并且不会给出预期的结果1.0
。是否可以编写另一个函数fix_singularity
在应用于上述函数时给出所需的结果,以便
fix_singularity(F)(0.0) == 1.0
或者正式fix_singularity
应该通过以下测试:
import numpy as np
def test_fix_singularity():
F = lambda x: np.sin(x)/x
x = np.array((0.0, pi))
np.testing.assert_array_almost_equal( F(x), [nan, 0] )
np.testing.assert_array_almost_equal( fix_singularity(F)(x), [1, 0] )
一种可能的实现是
def fix_singularity(F):
""" Fix the singularity of function F(x) """
def L(x):
f = F(x)
i = np.isnan(f)
f[i] = F(x[i] + 1e-16)
return f
return L
有没有更好的方法来做到这一点?
编辑:另外我怎样才能抑制警告:
Warning: invalid value encountered in divide