Предположим, у вас есть такая функция, как
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