【发布时间】:2011-07-30 05:44:55
【问题描述】:
假设你有一个类似
的函数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
【问题讨论】: