Desarrollo de Python
  1. Desarrollo de Python
  3. sympy.solve () no ofrece una de las soluciones con LambertW
Intereting Posts
Complejidad del operador * in * en Python Cómo usar el módulo timeit problema con las nuevas líneas cuando uso toprettyxml () Ejecutar progtwigs externos de Python con Eclipse PyDev ¿Cómo resolver la advertencia “MatplotlibDeprecationWarning: scipy.stats.norm.pdf”? parte del centro de recorte de una imagen numpy Django rest de correo electrónico de autenticación en lugar de nombre de usuario Cómo / por qué usamos operator.abs Python, escribiendo código de múltiples líneas en IDLE Python 2.x versiones múltiples problemas con PYTHONPATH La consulta de la base de datos de Python MySQL Connector con% s falla Hacer una clase de puntos en Python subproceso llamada ffmpeg (línea de comando) nbytes y getsizeof devuelven diferentes valores IDE agradable con el diseñador GUI para wxPython o Tkinter

sympy.solve () no ofrece una de las soluciones con LambertW

Fondo:
Estoy tratando de implementar una función haciendo una transformación inversa de muestreo . Uso sympy para calcular CDF y obtener su función inversa. Mientras que para algunos PDF simples obtengo resultados correctos, para un PDF cuya función inversa de CDF incluye la función de Lambert-W , los resultados son incorrectos.

Ejemplo:
Considere el siguiente ejemplo de CDF: 

import sympy as sym y = sym.Symbol('y') cdf = (-y - 1) * sym.exp(-y) + 1 # derived from `pdf = x * sym.exp(-x)` sym.plot(cdf, (y, -1, 5))

introduzca la descripción de la imagen aquí
Ahora calculando el inverso de esta función: 

 x = sym.Symbol('x') inverse = sym.solve(sym.Eq(x, cdf), y) print(inverse)

Salida: 

 [-LambertW((x - 1)*exp(-1)) - 1]

Esto, de hecho, es solo una twig izquierda de las y negativas de un CDF dado: 

 sym.plot(inverse[0], (x, -0.5, 1))

introduzca la descripción de la imagen aquí

Pregunta: ¿Cómo puedo obtener la twig correcta para las y positivas de un CDF dado?

Lo que intenté:

  1. Especificando x es solo positivo: 

     x = sym.Symbol('x', positive=True) y = sym.Symbol('y', positive=True)

    Esto no tiene ningún efecto, incluso para la primera gráfica de la FCD.

  2. Haciendo de CDF una función a Piecewise : 

     cdf = sym.Piecewise((0, y < 0), ((-y - 1) * sym.exp(-y) + 1, True))

    De nuevo sin efecto. Lo extraño aquí es que en otra computadora, al trazar esta función se obtuvo un gráfico adecuado con cero para las y negativas, pero la resolución de una twig positiva de y no funciona en ninguna parte. (¿Versiones diferentes? También tuve que especificar adaptive=False to sympy.plot para que funcione allí).

  3. Usando sympy.solveset lugar de sympy.solve :
    Esto solo da un ConditionSet(y, Eq(x*exp(y) + y - exp(y) + 1, 0), Complexes(S.Reals x S.Reals, False)) inútil ConditionSet(y, Eq(x*exp(y) + y - exp(y) + 1, 0), Complexes(S.Reals x S.Reals, False)) como resultado. Aparentemente, solveset aún no sabe cómo lidiar con LambertW funciones de LambertW . De los documentos:

    Cuando los casos no se resuelven o solo se pueden resolver de manera incompleta, se utiliza un ConditionSet y actúa como un objeto de Solveset sin evaluar. Todavía hay algunas cosas que solveset no puede hacer, lo que puede solve el antiguo, como resolver ecuaciones de tipo multivariable y LambertW no lineales.

¿Es un error o me falta algo? ¿Hay alguna solución para obtener el resultado deseado?

La inversa producida por sympy es casi correcta. El problema radica en el hecho de que la función LambertW tiene varias twigs sobre el dominio (-1 / e, 0). De forma predeterminada, utiliza la twig superior, sin embargo, para su problema, necesita la twig inferior. Se puede acceder a la twig inferior pasando un segundo argumento a LambertW con un valor de -1. 

 inverse = -sym.LambertW((x - 1)*sym.exp(-1), -1) - 1 sym.plot(inverse, (x, 0, 0.999))

Da

introduzca la descripción de la imagen aquí