Python: Wie vermeiden Sie numpy RuntimeWarning in Funktionsdefinition?

Question

Ich entwarf eine einfache Funktion, um eine mathematische Funktion zurückzugeben, die verwendet werden kann, um experimentelle Daten daran anzupassen. Die Funktionen sieht ziemlich ähnlich wie folgt aus:

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

Leider in Schwierigkeiten mit RunTimeWarnings wie ich laufen:

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

aufgrund Werte, die zu groß oder klein sind. Ich bin jedoch nicht in der Lage, dieses Problem alleine zu lösen. Gibt es eine Möglichkeit, meine Funktion neu zu definieren, so dass sie ohne Warnungen passieren wird?

Answer 1

Verwenden numpy.seterr zu kontrollieren, was numpy in diesem Umstand macht: http://docs.scipy.org/doc/numpy/reference/generated/numpy.seterr.html

die Warnungen Modul können Sie steuern, wie Warnungen sind oder nicht dargestellt werden: http://docs.python.org/library/warnings.html

Answer 2

Sie können numpy.errstate verwenden, welcher Kontext ein eingebauter, ist Manager. Dadurch können Sie festlegen, dass die Fehlerübergabe im Kontext der with-Anweisung steht.

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

Ich hatte dies vor kurzem zu tun, wenn für einige Legacy-Python-Code Unit-Tests zu schreiben.

Answer 3

Um dies zu umgehen, könnten Sie die Genauigkeit erhöhen, indem Sie den Typ der Array-Elemente ändern, auf denen Sie Ihre Funktion aufrufen.

Wenn beispielsweise Array Multiplizieren a mit großen Zahlen als Elemente von einem großen Gleitkommazahl löst eine Exception

RuntimeWarning: overflow encountered in multiply 

dann auf folgende Angaben:

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

keine Warnung erfolgt.