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

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))

不幸的是，我遇到了运行时警告的问题：

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)

没有警告。