Python numpy seterrcall用法及代码示例

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

用法:

numpy.seterrcall(func)

设置浮点错误回调函数或日志对象。

有两种方法可以捕获浮点错误消息。首先是使用 seterr 将 error-handler 设置为 ‘call’。然后，设置要使用此函数调用的函数。

第二个是使用 seterr 将 error-handler 设置为 ‘log’。浮点错误然后触发对所提供对象的‘write’ 方法的调用。

参数：

func： 可调用的 f(err, flag) 或带有 write 方法的对象

调用浮点错误的函数(‘call’-mode)或使用‘write’方法记录此类消息的对象(‘log’-mode)。

call 函数有两个参数。第一个是说明错误类型的字符串(例如“divide by zero”、“overflow”, “underflow” 或“invalid value”)，第二个是状态标志。该标志是一个字节，其四个最低有效位指示错误类型，“divide”, “over”, “under”, “invalid” 之一：

[0 0 0 0 divide over under invalid]

换句话说，flags = divide + 2*over + 4*under + 8*invalid。

如果提供了一个对象，它的 write 方法应该接受一个参数，一个字符串。

返回：

h： 可调用，日志实例或无

旧的错误处理程序。

例子：

错误回调：

>>> def err_handler(type, flag):
...     print("Floating point error (%s), with flag %s" % (type, flag))
...

>>> saved_handler = np.seterrcall(err_handler)
>>> save_err = np.seterr(all='call')

>>> np.array([1, 2, 3]) / 0.0
Floating point error (divide by zero), with flag 1
array([inf, inf, inf])

>>> np.seterrcall(saved_handler)
<function err_handler at 0x...>
>>> np.seterr(**save_err)
{'divide': 'call', 'over': 'call', 'under': 'call', 'invalid': 'call'}

记录错误信息：

>>> class Log:
...     def write(self, msg):
...         print("LOG: %s" % msg)
...

>>> log = Log()
>>> saved_handler = np.seterrcall(log)
>>> save_err = np.seterr(all='log')

>>> np.array([1, 2, 3]) / 0.0
LOG: Warning: divide by zero encountered in true_divide
array([inf, inf, inf])

>>> np.seterrcall(saved_handler)
<numpy.core.numeric.Log object at 0x...>
>>> np.seterr(**save_err)
{'divide': 'log', 'over': 'log', 'under': 'log', 'invalid': 'log'}