python function definition to solve equations

Asked Oct 09 '22 at 20:40

Active Oct 09 '22 at 20:40

Viewed 50 times

def fun(z):
   g = np.zeros(4 * N)
   for i in range(4 * N):
       f = np.zeros(4 * N)
       for j in range(4 * N):
            if z[j] != 0:
                f[i] = f[i] + M_kk[i][j] * z[j] * np.arcsinh(w_c / z[j])
            else:
                f[i] = f[i]
       g[i] = f[i] - z[i]
       return(g)

z = fsolve(fun,np.zeros(4 * N))

I want to define a function with 4*N equations. Do I put the return(g) at the right position in the end?

asked Oct 09 '22 at 20:40

karren lim

1

You almost never want to have `return` in a loop like that: the loop only runs through one iteration – Mad Physicist Oct 09 '22 at 20:42
maybe you want a `yield g` ... its hard to tell though – Joran Beasley Oct 09 '22 at 20:49

python function definition to solve equations

0 Answers0