Numpy Beginner's Guide (Update)

Time for action – fitting to a sine

In the previous Time for action section, we created a simple filter for detrended data. Now, let's use a more restrictive filter that will leave us only with the main frequency component. We will fit a sinusoidal pattern to it and plot our results. This model has four parameters—amplitude, frequency, phase, and vertical offset.

1. Define a residuals function based on a sine wave model:

   def residuals(p, y, x):
      A,k,theta,b = p
      err = y-A * np.sin(2* np.pi* k * x + theta) + b
      return err

2. Transform the filtered signal back to the original domain:

   filtered = -fftpack.irfft(fftpack.ifftshift(amps))

3. Guess the values of the parameters of which we are trying to estimate a transformation from the time domain into the frequency ...