Python numpy errstate用法及代码示例

本文简要介绍 python 语言中 numpy.errstate 的用法。

用法:

class  numpy.errstate(**kwargs)

用于浮点错误处理的上下文管理器。

使用 errstate 的实例作为上下文管理器允许该上下文中的语句以已知的错误处理行为执行。进入上下文后，错误处理将使用 seterr 和 seterrcall 设置，退出时将重置为之前的状态。

参数：

kwargs： {除，超过，低于，无效}

关键字参数。有效的关键字是可能的浮点异常。每个关键字都应该有一个字符串值，用于定义特定错误的处理方式。可能的值为 {‘ignore’, ‘warn’, ‘raise’, ‘call’, ‘print’, ‘log’}。

注意：

有关浮点异常类型和处理选项的完整文档，请参阅 seterr 。

例子：

>>> olderr = np.seterr(all='ignore')  # Set error handling to known state.

>>> np.arange(3) / 0.
array([nan, inf, inf])
>>> with np.errstate(divide='warn'):
...     np.arange(3) / 0.
array([nan, inf, inf])

>>> np.sqrt(-1)
nan
>>> with np.errstate(invalid='raise'):
...     np.sqrt(-1)
Traceback (most recent call last):
  File "<stdin>", line 2, in <module>
FloatingPointError: invalid value encountered in sqrt

在上下文之外，错误处理行为没有改变：

>>> np.geterr()
{'divide': 'ignore', 'over': 'ignore', 'under': 'ignore', 'invalid': 'ignore'}