python - 当值未使用时，如何避免“RuntimeWarning”除以零？

Question

我需要编写一个函数来计算(exp(x) - 1) / x，同时考虑到函数越x接近这一事实。我有以下实现01

def f(x):
    return np.where(x == 0, 1.0, np.expm1(x) / x)

这是正确的，因为它返回正确的值，但令人讨厌的是，当使用x包含零的参数调用它时，它会发出RuntimeWarning: invalid value encountered in true_divide. 我知道这是因为除以零实际上发生在第三个参数中，where尽管这些nan值从未使用过。

有没有更好的实现或方法来避免烦人的警告？

标签： pythonnumpy