encontrar cero de una funcion

[efirvida]

Hola a todos tengo una funcion a la cual le quiero hallar su raiz, estoy usando fsolve de scipy, pero cuando la ejecuto me da el siguiente error:

TypeError: fsolve: there is a mismatch between the input and output shape of the 'func' argument 'my_function'

pueden ayudarme aqui le spongo todo el codigo que tengo
saludos y gracias de antemano

Código Python:

def my_function(fi):
    from math import pi,sqrt,tan,sin,cos
    from numpy import array, radians
 
    m = 2.5
    z = 22
    beta = radians(16.)
    alfa = radians(20.)
    x = 0.
    c = 0.25
    f = 1.
    fr = 0.4
 
    rp = (m*z/cosd(beta))/2.
    ro = rp - m * (f + c - x)
    so = 2. * ro * (((pi/2. + 2.*x*tan(alfa))/z)+(tan(alfa)-alfa))
 
    l = pi/4. * m
    he = (f + c) * m
    rc = fr * m
    s = x * m
    d = he - rc - s
    PA = l + (rc/cos(alfa)) + (he -rc)*tan(alfa)
 
    Gama = radians(PA/rp)
    Beta = radians(so / (2.*ro))
 
    fi_e = radians(fi[0])
    fi_t = radians(fi[1])
 
    E1 = ro * (cos(Gama - Beta) + sin(Gama - Beta) * fi_e )
    E2 = -ro * (sin(Gama - Beta) - cos(Gama - Beta) * fi_e )
 
    T1 = rp - d *( l + rc/sqrt(d**2 + (rp * fi_t)**2))
    T2 = rp * fi_t *( l + rc/sqrt(d**2 + (rp * fi_t)**2))
    
    return E1 * cos(fi_e) + E2 *sin(fi_e) - T1 *cos(fi_t) + T2 * sin(fi_t)
     
from scipy.optimize import fsolve
from math import pi
 
print  fsolve(my_function, [-pi/2, 0])
 

[razpeitia]

Lo que hace fsolve es sencillo, resuelve un sistema de ecuaciones no lineales.

Ahora, si tu le pasas un arreglo de 2 elementos fsolve espera que la función que le pases (my_fun) devuelva un arreglo de 2 elementos. Esto es porque el segundo parámetro que le pasas son los puntos en el eje X de donde quieres que empiece cada ecuación en el sistema de ecuaciones. [-pi/2, 0]

Mira este link en la sección 1.2 ejemplo 4 y 5 https://sites.google.com/a/aims-sene...tial-equations

No puedo arreglar tu script porque no se lo que tratas de hacer.

Si tienes un sistema de ecuaciones de 2 ecuaciones o si tienes una sola ecuación y quieres que empiece en -pi/2.

Si es el caso del sistema de ecuaciones de 2 ecuaciones entonces obviamente te falta devolver una lista de tamaño 2, donde el primer item es la evaluación de tu primera ecuación y el segundo es la evaluacion de tu segunda ecuación.

Si solo tienes una sola ecuación entonces no veo porque envías una lista con 2 elementos. Y porque ademas haces uso de los 2 argumentos.

Si nos pusieras el sistema de ecuaciones que estas tratando de resolver te podríamos ayudar a implementarlo.

[Bael_Balzac]

En el código que muestras veo un cosd, es recomendable importar numpy asi:

Código Python:

import numpy as np
 
ejemplo1 = np.pi
ejemplo2 = np.sqrt(2.0)
ejemplo3 = np.cos(0.25)
 

Tambien puedes usar los solucionadores numéricos y simbólicos de sympy.
Revisando la documentación de scipy veo que si quieres hallar las raices debes usar:

Código Python:

from scipy.optimize import root