diff --git a/endaq/calc/shock.py b/endaq/calc/shock.py index 4328f1e..60854a9 100644 --- a/endaq/calc/shock.py +++ b/endaq/calc/shock.py @@ -490,7 +490,7 @@ def shock_spectrum( ) srs[var] = pivot.max(axis=1) srs.columns.name = accel.columns.name - return srs + return srs[["Resultant"]] if aggregate_axes == True else srs omega = 2 * np.pi * freqs