def noise_ar1(n, rows=1, a1=0.9): x = noise_white(n, rows=rows) for row in x: row[:] = fftconvolve(row, np.array([1., a1]), mode='same') x = (x.T/np.sqrt(np.mean(x**2, axis=1))).T return x