NumPy shading of plot based on number of iterations required to converge to a root

Question

I have a function, which_root_z4(z,n), which creates and returns array consisting of values 0, 1, 2, 3 corresponding to the indices of the zeroth,