如何查看牛顿法运行的迭代次数？

Question

所以基本上我想获取我的牛顿法找到根所需的迭代次数，然后将这个数字应用到我的配色方案中，使迭代次数越长，颜色越深，并且越少，颜色越饱满。

这是我的代码

from numpy import *
import pylab as pl
def myffp(x):
    return x**3 - 1, 3*(x**2)    
def newton( ffp, x, nits):
    for i in range(nits):
        #print i,x
        f,fp = ffp(x)
        x = x - f/fp
    return x    
q = sqrt(3)/2
def leggo(xmin=-1,xmax=1,jmin=-1,jmax=1,pts=1000,nits=30):
    x = linspace(xmin, xmax, pts)
    y = linspace(jmin, jmax, pts)*complex(0,1)
    x1,y1 = meshgrid(x,y)                   
    n = newton(myffp,x1+y1,nits)                  #**here is where i wanna see the number of iterations newton's method takes to find my root** 
    r1 = complex(1,0)  
    r2 = complex(-.5, q)
    r3 = complex(-.5,-q)
    data = zeros((pts,pts,3))   
    data[:,:,0] = abs(n-r1)         #**and apply it here**
    data[:,:,2] = abs(n-r2)
    data[:,:,1] = abs(n-r3)
    pl.show(pl.imshow(data))    
leggo() 

主要问题是找到迭代次数，然后我可以弄清楚如何将其应用于使颜色变暗，但现在它只是找到通过牛顿方法运行的每个值所需的迭代次数。

Answer 1

也许最简单的方法是重构您的 newton 函数，以便它跟踪总迭代，然后返回它（当然还有结果），例如，

def newton( ffp, x, nits):
    c = 0                   # initialize iteration counter
    for i in range(nits):
        c += 1              # increment counter for each iteration 
        f, fp = ffp(x)
        x = x - f/fp
    return x, c             # return the counter when the function is called

所以在您的代码主体中，更改您对 newton 的调用，如下所示：

res, tot_iter = newton(myffp, x, nits)

最后一次调用 newton 的迭代次数存储在 tot_iter

---

顺便说一句，您对牛顿法的实现似乎不完整。

例如，它缺少针对某些收敛标准的测试。

这是一个在 python 中有效的简单实现：

def newtons_method(x_init, fn, max_iter=100):
    """
    returns: approx. val of root of the function passed in, fn;
    pass in: x_init, initial value for the root; 
    max_iter, total iteration count not exceeded;
    fn, a function of the form: 
    def f(x): return x**3 - 2*x
    """
    x = x_init
    eps = .0001
    # set initial value different from x_init so at lesat 1 loop
    x_old = x + 10 * eps        
    step = .1
    c = 0
    # (x - x_old) is convergence criterion
    while (abs(x - x_old) > eps) and (c < max_iter):
        c += 1
        fval = fn(x)
        dfdx = (fn(x + step)) - fn(x) / step
        x_old = x
        x = x_old - fval / dfdx
    return x, c

Answer 2

您当前用于newton() 的代码具有固定的迭代次数（nits - 以 30 的形式传入），因此结果会有些琐碎且无趣。

您似乎正在尝试生成牛顿分形 - 您尝试使用的方法不正确；典型的着色模式是基于函数的输出，而不是迭代次数。完整说明请参见the Wikipedia article。