diff options
| author | Eric Wieser <wieser.eric@gmail.com> | 2018-07-30 22:52:14 -0700 |
|---|---|---|
| committer | Eric Wieser <wieser.eric@gmail.com> | 2018-07-30 22:52:14 -0700 |
| commit | dc6b8df8d1025c522a2877bb06547d143f481cc6 (patch) | |
| tree | 39425cad8b1b676de0af66e914a15bd5c53a2c70 /numpy/lib/function_base.py | |
| parent | 083aedba2f49e1b490d3e122f5927f0718cf202c (diff) | |
| download | numpy-dc6b8df8d1025c522a2877bb06547d143f481cc6.tar.gz | |
MAINT: np.angle: Remove unnecessary multiplication
Diffstat (limited to 'numpy/lib/function_base.py')
| -rw-r--r-- | numpy/lib/function_base.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py index 1a43da8b0..ea6339514 100644 --- a/numpy/lib/function_base.py +++ b/numpy/lib/function_base.py @@ -1339,10 +1339,6 @@ 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 @@ -1350,7 +1346,11 @@ def angle(z, deg=0): 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): |