求教时间-小波能量谱

ziyou2050

求教各位高手怎么画时间-小波能量谱图，我是新手，自己写了个，抛砖引玉，希望各位高手指教
load 导入数据
n=0:N-1;t=n/fs;   %时间序列
c=cwt(x,1:10,'morl');%小波变换，尺度1:10，小波morl,结果放入c中
c=c.^2;
s1=sum(c,1);%同一时间参数下各尺度的小波系数平方相加
figure(1);plot(t,s1);    %时间--小波能量谱
xlabel('时间 t/s');ylabel('小波能量谱');

y=fft(s1);  %对信号进行快速Fourier变换
mag=abs(y)*2/N;     %求得Fourier变换后的振幅
mag(1)=0;
f=n*fs/N;    %如何求频率序列
figure(2);plot(f(1:N/2),mag(1:N/2));   %绘出随频率变化的振幅
xlabel('频率/Hz');
ylabel('振幅');title('频域图');
grid on;%图添加网格线