在python上的循环内求解非线性方程

Question

我需要一些帮助来更有效地编写代码。鉴于某些数组中的元素，以下代码旨在解决变量hi的非线性方程。例如：

import numpy  as np 
import tensorflow as tf
import scipy.optimize 

kon = 0.01
mu  = 1.5
fi  = 0.5 
kappa = 22

n =100
xs = tf.random.normal(shape=(n,), stddev=0.2)
eps = tf.random.normal(shape=(n,), stddev=0.17)
z = tf.sigmoid(tf.random.normal(shape=(n,), stddev=0.22))

# non-linear equation to solve for leisure 
def F(hi):
  return (mu/fi)*np.log(hi) -(1-mu)*kappa*(hi)**(1+(1/fi))-mu*(np.log(w*ei*xs)-np.log(kon))-np.log(ze)
hvec = np.empty((0,))
# leisure today 
for ze,ei,xs in zip(z, eps, x0):
    ei=tf.exp(ei)
    xs=tf.exp(xs)
    htemp = scipy.optimize.newton_krylov(F, 0.5)
    hvec = np.append(hvec, htemp)   

z，eps，x0是大小为n（例如n=100）的数组

是否有更有效的方法来做同样的事情？例如，通过避免循环？

我基本上想为这三个数组中的每个元素求解hi。