Remove singleton dimension in a python2.6 compatible way

scipy · May 5, 2015 · 2a5dbad · 2a5dbad
1 parent 1f8bd40
commit 2a5dbad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scipy/signal/spectral.py b/scipy/signal/spectral.py
@@ -368,7 +368,7 @@ def csd(x, y, fs=1.0, window='hanning', nperseg=256, noverlap=None, nfft=None,
         if Pxy.shape[-1] > 1:
             Pxy = Pxy.mean(axis=-1)
         else:
-            Pxy = np.squeeze(Pxy, axis=-1)
+            Pxy = np.reshape(Pxy, Pxy.shape[:-1])
 
     return freqs, Pxy