python - 如何在保持所有其他警告正常的同时摆脱 python 中的特定警告消息？

Question

我在 python 脚本中隐性地做一些简单的数学运算，并收到以下警告:

"Warning: divide by zero encountered in divide".

为了提供一些上下文，我采用两个值并试图找到值 (a - b)/a 的百分比差异，如果它高于某个范围，则处理它，但有时a 或 b 的值为零。

我想摆脱这个特定的警告(在特定的行)，但到目前为止我发现的所有信息似乎都向我展示了如何停止所有警告(我不想要)。

以前写 shell 脚本的时候，我可以做这样的事情

code...
more code 2 > error.txt
even more code  

在该示例中，我会收到“代码”和“更多代码”命令的警告，但不会收到第二行的警告。

这可能吗？

Answer 1

如果 Scipy 使用 warnings 模块，那么您可以禁止显示特定警告。在程序开始时试试这个:

import warnings
warnings.filterwarnings("ignore", message="divide by zero encountered in divide")

如果您希望这仅适用于一段代码，请使用警告上下文管理器:

import warnings
with warnings.catch_warnings():
    warnings.filterwarnings("ignore", message="divide by zero encountered in divide")
    # .. your divide-by-zero code ..