z^5 1的吸引盆（Matlab）

Question

我试图画出复杂方程z^5 1=0的吸引盆。我已经设法制作了一个大矩阵（1000000个元素），只有6个不同的值，如果等值点收敛，则有5个根，如果发散，则有其他的根。但我无法将每种不同类型的值映射到不同的颜色。到目前为止，我的代码是

    N=1000;
    x=linspace(-5,5,N);
    y=linspace(-5,5,N); 
    A=ones(N,1)*x + 1i*y'*ones(1,N); %A is a 1000x1000 matrix.


    f=@(x) x.^5+1; 
    fp=@(x) 5*x.^4;

    for i=1:20
    A=A-f(A)./fp(A); 
    end

PColor似乎不适用于复杂的参数，如果我把abs（A）作为参数，所有不同的根转换为相同的数字，我最终得到一个只有一种颜色的图形，这很奇怪，因为有些点分歧，所以我应该有2种颜色。有什么想法我怎么能绘制这个复杂的数组？

Answer 1

不要计算复数的大小，而是使用角度计算它们的相位，这对于每个解决方案都是不同的，例如：

h = pcolor(x, y, angle(A));
set(h, 'EdgeColor', 'none');