[Numpy-discussion] How to remove fortran-like loops with numpy?
Robert Kern
robert.kern at gmail.com
Mon Jun 8 18:25:37 EDT 2009
On Mon, Jun 8, 2009 at 17:04, David Goldsmith<d_l_goldsmith at yahoo.com> wrote:
>
> I look forward to an instructive reply: the "Pythonic" way to do it would be to take advantage of the facts that Numpy is "pre-vectorized" and uses broadcasting, but so far I haven't been able to figure out (though I haven't yet really buckled down and tried real hard) how to broadcast a conditionally-terminated iteration where the number of iterations will vary among the array elements. Hopefully someone else already has. :-)
You can't, really. What you can do is just keep iterating with the
whole data set and ignore the parts that have already converged. Here
is an example:
z = np.zeros((201,201), dtype=complex)
Y, X = np.mgrid[1:-1:-201j, -1.5:0.5:201j]
c = np.empty_like(z)
c.real = X
c.imag = Y
N = np.zeros(z.shape, dtype=int)
while ((N<30) | (abs(z)<2)).all():
N += abs(z) < 2
z = z ** 2 + c
N[N>=30] = 0
--
Robert Kern
"I have come to believe that the whole world is an enigma, a harmless
enigma that is made terrible by our own mad attempt to interpret it as
though it had an underlying truth."
-- Umberto Eco
More information about the NumPy-Discussion
mailing list