dedebenui/scgenerator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

change: more robust g12

Browse Source
This commit is contained in:
Benoît Sierro
2023-08-18 14:22:06 +02:00
parent 313bb3295e
commit 0cd43802e5
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/scgenerator/math.py
View File

@@ -689,11 +689,8 @@ def mean_angle(values: np.ndarray, axis: int = 0):
array([ 0.92387953+0.38268343j, 0.28978415+0.95709203j, -1. +0.j ]) array([ 0.92387953+0.38268343j, 0.28978415+0.95709203j, -1. +0.j ])
""" """
new_shape = values.shape[:axis] + values.shape[axis + 1 :] values = np.divide(
total_phase = np.sum( values, np.abs(values), out=np.zeros(values.shape, dtype=values.dtype), where=values != 0
values / np.abs(values),
axis=axis,
where=values != 0,
out=np.zeros(new_shape, dtype="complex"),
) )
total_phase = np.sum(values, axis=axis)
return (total_phase) / np.abs(total_phase) return (total_phase) / np.abs(total_phase)