Integer divisions when possible. No need to cast to float anymore.

Author: arokem

nitime/algorithms/spectral.py

@@ -105,7 +105,7 @@ def get_spectra(time_series, method=None):
         Fs = method.get('Fs', 2 * np.pi)
         detrend = method.get('detrend', mlab.detrend_none)
         window = method.get('window', mlab.window_hanning)
-        n_overlap = method.get('n_overlap', int(np.ceil(NFFT / 2)))
+        n_overlap = method.get('n_overlap', int(np.ceil(NFFT //  2)))
 
         # The length of the spectrum depends on how many sides are taken, which
         # depends on whether or not this is a complex object:
@@ -587,7 +587,7 @@ def multi_taper_psd(
     sdf_est /= Fs
 
     if sides == 'onesided':
-        freqs = np.linspace(0, Fs / 2, NFFT / 2 + 1)
+        freqs = np.linspace(0, Fs / 2, NFFT //  2 + 1)
     else:
         freqs = np.linspace(0, Fs, NFFT, endpoint=False)
 
@@ -726,7 +726,7 @@ def multi_taper_csd(s, Fs=2 * np.pi, NW=None, BW=None, low_bias=True,
     csdfs /= Fs
 
     if sides == 'onesided':
-        freqs = np.linspace(0, Fs / 2, NFFT / 2 + 1)
+        freqs = np.linspace(0, Fs / 2, NFFT //  2 + 1)
     else:
         freqs = np.linspace(0, Fs, NFFT, endpoint=False)
 

nitime/utils.py

@@ -493,7 +493,7 @@ def adaptive_weights(yk, eigvals, sides='onesided', max_iter=150):
         """)
         # we'll hope this is a correct length for L
         N = yk.shape[-1]
-        L = N / 2 + 1 if sides == 'onesided' else N
+        L = N // 2 + 1 if sides == 'onesided' else N
         return (np.multiply.outer(np.sqrt(eigvals), np.ones(L)), 2 * K)
     rt_eig = np.sqrt(eigvals)
 
@@ -1206,10 +1206,10 @@ def fftconvolve(in1, in2, mode="full", axis=None):
 # 'get' utils
 #-----------------------------------------------------------------------------
 def get_freqs(Fs, n):
-    """Returns the center frequencies of the frequency decomposotion of a time
+    """Returns the center frequencies of the frequency decomposition of a time
     series of length n, sampled at Fs Hz"""
 
-    return np.linspace(0, float(Fs) / 2, float(n) / 2 + 1)
+    return np.linspace(0, Fs / 2, n / 2 + 1)
 
 
 def circle_to_hz(omega, Fsamp):