在 Python 中完全禁用/替换 ZeroDivisionError

Question

美好的一天。 我一直在搜索相关帖子，但没有找到我想找到的理想解决方案。让我描述一下我的问题：

我正在分析语料库中的文本，并从这些文本中提取特征，然后将这些特征存储在一个数组中。其中一些特征涉及比例，例如男性代词“他”与女性代词“她”的比例。问题是，对于某些变量，值将为零，它们将引发 ZeroDivisionError。

由于我计算了大约 100 个这些比率，因此在每个比率计算周围都包含一个 try/catch 异常听起来太麻烦了。

我发现我可以做到

#16,RATIO_masculine_femenine
feature_map.append(numOfHe / numOfShe if numOfShe else 0)

但这还是有点太费力了。 我想知道是否有办法在脚本的开头声明任何 ZeroDivisionError 都应该用 NaN 或 0 或任何其他可能适合的值代替。

谢谢

Answer 1

pythonic 的答案是将其包装在一个函数中，例如：

def ratio(a, b):
    if b == 0:
        return 0
    else:
        return a / b

feature_map.append(ratio(numOfHe, numOfShe))

函数的确切形式取决于您的其余代码，但如果您要编写数百次这样的行，那么您可能应该将它包装在一个函数中，或者至少使用一个循环。此外，numOfHe 和 numOfShe 之类的变量名称暗示您可能更适合使用 dict。

更新

我从您的代码链接中看到每个 calc 实际上是完全不同的，因此您可能无法轻松循环它。由于计算仍然相对简单，您可以尝试使用 eval 的技巧，如下所示：

calcs = [
    ...
    (12, 'h + ha + hw + hy'),
    (13, '(h + ha) / (hw + hy)'),
    ...
]

for index, calc in calcs:
    try:
        v = eval(calc, locals())
    except ZeroDivisionError:
        v = 0
    feature_map.append(v)

您还可以将其他信息添加到calcs，并改用namedtuple。如果有帮助的话，您也可以使用类来动态评估计算。

Answer 2

如果您将 int 对象包装在自定义子类中，则可以处理一次：

class SafeInt(int):
    def __div__(self, y):
        try:
            return SafeInt(super(SafeInt, self).__div__(y))
        except ZeroDivisionError:
            return SafeInt(0)

覆盖所有 ints:

original_int = int
int = SafeInt
int(5) / 0
# O: 0

覆盖一些 ints:

SafeInt(5) / 0
# O: 0

您必须小心保持对象为 SafeInt。你会注意到我在__div__ 中返回的所有内容都包含在SafeInt() 中。 int 对象是不可变的，您每次都必须显式返回一个新的 SafeInt 对象。这意味着您可能需要为SafeInt() 中的每个函数制作一个装饰器以确保这一点。我把它作为练习留给读者！

否则你会得到这样的结果：

>>> SafeInt(5) / 0
0   # this is a SafeInt object
>>> _ / 0
0   # this is a SafeInt object; no error
>>> SafeInt(5) + 0
5   # this is a basic int object
>>> _ / 0
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ZeroDivisionError: integer division or modulo by zero

最后一点：您可以将SafeInt 作为defaultdict 的参数传递给所有成员SafeInt！

---

编辑：知道你希望它发生在 all ints，我希望这样的事情可能会起作用，但它是不允许的（有充分的理由）：

>>> def wrapdiv(olddiv):
...     def newdiv(self, y):
...         try:
...             olddiv(self, y)
...         except ZeroDivisionError:
...             return 0
...     return newdiv
...
>>> int.__div__ = wrapdiv(int.__div__)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: can't set attributes of built-in/extension type 'int'