MAINT: np.angle: Remove unnecessary multiplication

eric-wieser · eric-wieser · commit dc6b8df8d102 · 2018-07-30T22:52:14.000-07:00
diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py
@@ -1339,18 +1339,18 @@ def angle(z, deg=0):
     45.0
 
     """
-    if deg:
-        fact = 180/pi
-    else:
-        fact = 1.0
     z = asarray(z)
     if (issubclass(z.dtype.type, _nx.complexfloating)):
         zimag = z.imag
         zreal = z.real
     else:
         zimag = 0
         zreal = z
-    return arctan2(zimag, zreal) * fact
+
+    a = arctan2(zimag, zreal)
+    if deg:
+        a *= 180/pi
+    return a
 
 
 def unwrap(p, discont=pi, axis=-1):
