[Numpy-discussion] How to get rid of the loop?
Sat Nov 7 12:57:14 CST 2009
On Sat, Nov 7, 2009 at 1:51 PM, Stas K <firstname.lastname@example.org> 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.
> NumPy-Discussion mailing list
More information about the NumPy-Discussion