Erro ao resolver a equação da matriz com numpy
Estou escrevendo meu próprio algoritmo de Newton-Raphson em Python usandosympy
enumpy
.
O código está abaixo, mas você pode ignorá-lo e pular para o erro:
CÓDIGO
def newtonRhapson(fncList, varz, x0):
jacob = []
for fnc in fncList:
vec = []
for var in varz:
res = fnc.diff(var)
for i in range(len(varz)):
res = res.subs(varz[i], x0[i])
vec.append(res)
jacob.append(numpy.array(vec, dtype='float64'))
fx0=[]
for fnc in fncList:
res2 = fnc
for i in range(len(varz)):
res2 = res2.subs(varz[i], x0[i])
fx0.append(res2)
j = jacob
f = fx0
print j
print ''
print f
print numpy.linalg.solve(j,f).tolist()
Os argumentos para a função são:
fncList
- uma lista python de funções usandoSympy
símbolos
varz
- uma lista contendo esses símbolos (variáveis)
x0
- palpite inicial
ERRO
Até o ponto em queprint
j
ef
funciona bem e imprime o seguinte:
[array([-9.13378682, -5.91269838]), array([ 4.84401379, 1.01980286])]
[-5.15598620617611, 5.13378681611922]
quando eu corro:
newtonRhapson([5*cos(a)+6*cos(a+b)-10, 5*sin(a)+6*sin(a+b)-4], [a,b], [0.7,0.7])
Mas ao executar a linha:
print numpy.linalg.solve(j,f).tolist()
Eu recebo o erro:
File "/Users/me/anaconda/lib/python2.7/site- packages/numpy/linalg/linalg.py", line 384, in solve
r = gufunc(a, b, signature=signature, extobj=extobj)
TypeError: No loop matching the specified signature and casting was found for ufunc solve1