I am trying to attempt plot a real, imaginary, phase, and modulus of a wavelets coefficient. The mother wavelets i am using is morlet. Down below is my signal

dt = 0.001 sampling_rate = 1000 K = 3 t = np.arange(0, 1, dt) # Time from 0 to 1s f1 = np.sin(2np.pi50t) + np.sin(2np.pi100t)(t < 0.5) # From 0 to 0.5s f2 = np.sin(2np.pi20t) + np.sin(2np.pi100t)(t >= 0.5) + K*(1*(0.8 == t)+1*(0.804 == t)) I would like to ask that if what i am trying to do having any buit in function or it is a math problem that i have to write it out myself. I read document for pywt but they dont have any function that can plot real, imaginary or whatever above. Thank you