Numpy数组索引问题

Question

我正在尝试为多变量场景实现Newtons_method。我试图输出多次迭代，但我得到一个索引错误

import symengine
from symengine import var
import numpy as np
vars =var("x y")
sol=[]
def jacob():
    f = ['2*x+y**2-8','x**2-y**2+x*y-3'] # Define function
    J = symengine.zeros(len(f),len(vars)) # Initialise Jacobian matrix
    # Fill Jacobian matrix with entries
    for i, fi in enumerate(f):
        for j, s in enumerate(vars):
            J[i,j] = symengine.diff(fi, s)
            sol.append(J[i,j])
    a = np.array([sol[0],sol[1],sol[2],sol[3]])
    return a

def eval1(func,val1,val2):
    z=[]
    for b in func:
        b=str(b)
        x=val1
        y=val2
        z.append(eval(b))
    return np.array([[z[0],z[1]],[z[2],z[3]]])


def newtons_method(f, df, x0, e):
    q=10
    while q > e:
        q=q-1
        x0 = np.absolute(x0 - (np.array([[f(x0[0][0])],[df(x0[1][0])]]))/(eval1(jacob(),x0[0][0],x0[1][0])))
    print('Root is at: ', x0)
    print('f(x) at root is: ', f(x0))


def f(x):
    return 2*x+y**2-8

def df(x):
    return x**2-y**2+x*y-3
x0s = np.array([[1],[1]])

for x0 in x0s:
    newtons_method(f, df, x0, 1)

我收到错误：标量变量索引无效。我在newtons_method函数下将值设置为x0的地方，对于出了什么问题的想法？

Answer 1

当调用newtons_method函数时，[[1]，[1]]的元素，即[1]作为参数x0传递。在函数内部，使用了x0 [0] [0]，它给标量变量错误提供了无效索引。

如果要同时使用这两个元素，则应将x0s作为参数传递。

x0s = np.array([[1],[1]])
newtons_method(f, df, x0s, 1)