In [9]:

    

%load_ext autoreload
%autoreload 2

import waipy
import numpy as np


z = linspace(0,2048,2048)
x = sin(50*np.pi*z)+3.5*randn(size(z))
y = cos(50*np.pi*z+pi/4)+4*randn(size(z))

data_norm = waipy.normalize(x)
result = waipy.cwt(data_norm, 1, 1, 0.25, 4, 4/0.25, 0.72, 6, mother='Morlet',name='x')
waipy.wavelet_plot('Sine with noise', z, data_norm, 0.03125, result)

data_norm1 = waipy.normalize(y)
result1 = waipy.cwt(data_norm1, 1, 1, 0.25, 4, 4/0.25, 0.72, 6, mother='Morlet',name='y')
waipy.wavelet_plot('Cosine with noise', z, data_norm1, 0.03125, result1)

cross_power, coherence, phase_angle = waipy.cross_wavelet(result['wave'], result1['wave'])
waipy.plot_cross('Crosspower sine and cosine', cross_power, phase_angle, z, result, result1)


    

In [7]: