Recipe 577729 revision 1

# http://en.wikipedia.org/wiki/Lambert_W_function
# FB - 201105297
import math
eps = 0.00000001 # max error allowed
def w0(x): # Lambert W function using Newton's method
    w = x
    while True:
        ew = math.exp(w)
        wNew = w - (w * ew - x) / (w * ew + ew)
        if abs(w - wNew) <= eps: break
        w = wNew
    return w

# usage example: solution of x**x = a
a = float(raw_input("Enter a positive number: "))
x = math.log(a) / w0(math.log(a)) # solution 1
print x
x = math.exp(w0(math.log(a))) # solution 2
print x

• « Back to Recipe 577729

History

• revision 1 (12 years ago)