[Numpy-discussion] How to get rid of the loop?

josef.pktd@gmai... josef.pktd@gmai...
Sat Nov 7 12:57:14 CST 2009


On Sat, Nov 7, 2009 at 1:51 PM, Stas K <stanconn@gmail.com> wrote:
> Can I get rid of the loop in this example? And what is the fastest way
> to get v in the example?
>
> ar = array([1,2,3])
> for a in ar:
>    for b in ar:
>        v = a**2+b**2

>>> ar[:,None]**2 + ar**2
array([[ 2,  5, 10],
       [ 5,  8, 13],
       [10, 13, 18]])

I think, for this case there is also directly a function in numpy
hypot which should also work.

Josef
> _______________________________________________
> NumPy-Discussion mailing list
> NumPy-Discussion@scipy.org
> http://mail.scipy.org/mailman/listinfo/numpy-discussion
>


More information about the NumPy-Discussion mailing list