python：绕过零除

Question

我有一个浮点数的大数据集。我遍历它们并np.log(x)为它们中的每一个进行评估。我懂了

RuntimeWarning: divide by zero encountered in log

我想解决这个问题，如果发生此错误，则返回0。

我正在考虑定义一个新功能：

def safe_ln(x):
    #returns: ln(x) but replaces -inf with 0
    l = np.log(x)
    #if l = -inf:
    l = 0
    return l

基本上，我需要一种测试输出是否正确的方法，-inf但我不知道如何进行。感谢您的帮助！

Answer 1

由于logforx=0为负无穷大，因此我只需要检查输入值是否为零并返回您想要的值即可：

def safe_ln(x):
    if x <= 0:
        return 0
    return math.log(x)

编辑 ：小编辑：您应该检查所有小于或等于0的值。

编辑2 ：np.log当然是在numpy数组上计算的函数，对于单个值，您应该使用math.log。这是上面的函数与numpy的外观：

def safe_ln(x, minval=0.0000000001):
    return np.log(x.clip(min=minval))