Question

我设计了一个简单的函数来返回一个数学函数，该函数可用于将实验数据拟合到它。这些函数看起来非常类似于：

def colecole_2(f,*p):
    term1=p[0] * ( 1 - 1 / (1 + numpy.power((0+1j) * 2 * numpy.pi * f * p[1], p[2])))
    term2=p[3] * ( 1 - 1 / (1 + numpy.power((0+1j) * 2 * numpy.pi * f * p[4], p[5])))
    return p[6]*(1-abs( term1+ term2))

不幸的是，我遇到了RunTimeWarnings的麻烦：

RuntimeWarning: overflow encountered in power
RuntimeWarning: overflow encountered in divide

由于值太大或太小。我无法自己解决这个问题。有没有办法重新定义我的功能，所以它会在没有警告的情况下通过？

Answer 1

您可以使用numpy.errstate这是一个内置的上下文管理器。这将允许您将错误处理设置在with语句的上下文中。

import numpy
# warning is not logged here. Perfect for clean unit test output
with numpy.errstate(divide='ignore'):
    numpy.float64(1.0) / 0.0

最近我在编写一些遗留python代码的单元测试时不得不这样做。

Answer 2

使用numpy.seterr来控制numpy在这种情况下的作用：http://docs.scipy.org/doc/numpy/reference/generated/numpy.seterr.html

使用警告模块控制警告的显示方式：http://docs.python.org/library/warnings.html

Answer 3

要解决此问题，您可以通过修改调用函数的数组元素的类型来提高精度。

例如，如果将数组 a 与大数相乘作为元素乘以大的浮点数会引发异常

RuntimeWarning: overflow encountered in multiply

然后指定以下

a = np.array(a, dtype=np.float128)

没有发出警告。

Python：如何在函数定义中避免numpy RuntimeWarning？

3 个答案: