[Numpy-discussion] method to calculate the magnitude squared

[Phillip Feldman]

In communications and signal processing, it is frequently necessary to
calculate the power of a signal.  This can be done with a function like the
following:

def magsq(z):
   """
   Return the magnitude squared of the real- or complex-valued input.
   """
   return z.real**2 + z.imag**2

A high percentage of the scripts that I write contain or import a function
like this.  It would be great if there were a built-in method in NumPy,
preferably with a name like `magsq`, `mag2`, or `msq`.

Phillip
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/numpy-discussion/attachments/20150918/2a741a1f/attachment.html>