Вейвлеты

datafile = np.loadtxt("sin_3Hz_data.txt", float)
arrayofdatafile = np.array(datafile)
arrayofdatafile = arrayofdatafile.transpose()
times = arrayofdatafile[0, :]
datas = arrayofdatafile[1, :]
dt = 1 / 1000
scales = np.arange(1, 10, 0.1)
f = pywt.scale2frequency("gaus1", scales) / dt
[coefficients, frequencies] = pywt.cwt(datas, f, 'cmor1.0-0.4', dt)
 
power = (abs(coefficients)) ** 2
period = 1. / frequencies
 
fig, ax = subplots(figsize=(15, 10))
im = ax.contourf(times, scales, log2(power), extend='both',cmap="inferno")
 
show()

import numpy as np
import pywt
from matplotlib import pyplot as plt
times=np.arange(0, 20, 0.001)
datas=np.sin(3*2*np.pi*times)
 
dt = 1 / 1000
scales = np.arange(0.1, 10, 0.01)
f = pywt.scale2frequency("cmor7.0-1.0", scales) / dt
[coefficients, frequencies] = pywt.cwt(datas, f, 'cmor7.0-1.0', dt)
 
power = (np.abs(coefficients)) 
period = 1. / frequencies
 
fig, ax = plt.subplots(figsize=(15, 10))
#im = ax.contourf(times, scales, power, extend='both',cmap="gray")
im = ax.imshow( power, aspect='auto', origin='lower', extent=[0,20, 0.1, 10])
 
 
#plt.plot(times, datas)
plt.show()

def wavelet(i):
    datas = arrayofdatafile[:, i]
    [coefficients, frequencies] = pywt.cwt(datas, f, 'cmor7.0-1.0', dt)
    #print("Graph's start")
    #a = timeit.default_timer()
    power = (np.abs(coefficients))
    #fig, ax = plt.subplots(figsize=(15, 10))
    #im = ax.imshow(power, aspect='auto', origin='lower', extent=[0, l * dt, 1, 5])
    #plt.savefig('saved' + str(i) + '.png')
    #plt.close()
    #plt.show()
    #print("Graph's complete. Time is: " + str(timeit.default_timer() - a))
 
threads = []
a = timeit.default_timer()
for i in range(0, 27):
    threads.append(threading.Thread(target=wavelet, args=(i, )))
    threads[i].start()
    threads[i].join()
 
#print("Number of processed electrode is: " + str(i))
print("Wavelet's complete. Time is: " + str(timeit.default_timer() - a))
#plt.show()

for i in range(0, 27):
    threads.append(threading.Thread(target=wavelet, args=(i, )))
    threads[i].start() #старт всех потоков
 
for i in range(0, 27):
    threads[i].join()

f = pywt.scale2frequency("cmor7.0-1.0", scales) / dt

scales = np.arange(0.1, 10, 0.01)