前提・実現したいこと
時系列データとそれをウェーブレット変換を用いて周波数解析したスペクトログラムの出力をしたのですが、時系列データのグラフのx軸とカラーマップのx軸が異なってしまいました。
2つのx軸をそろえるためにはどのようにプログラムを変えるとよいでしょうか。
該当のソースコード
Python
1m=6 2#解析する周波数帯[Hz] 3list1 = range(1, 80, 1) 4 5print("start") 6for freq in list1: 7 print(freq,end="") 8 print('-',end="") 9 mw2=signal.morlet(int(fs/freq*m*2), m,1.0, complete=True) 10 corr_tmp=np.correlate(ags_data,mw2)/sum(abs(mw2)) 11 12 sup_first=np.zeros(int(len(mw2)/2)) 13 sup_end=np.zeros(len(ags_data)-len(corr_tmp)-int(len(mw2)/2)) 14 corr_tmp=np.append(sup_first,corr_tmp) 15 corr_tmp=np.append(corr_tmp,sup_end) 16 17 if freq==list1[0]: 18 corr_stack=abs(corr_tmp[::10])**2 19 else: 20 corr_stack=np.vstack((corr_stack, abs(corr_tmp[::10])**2)) 21print("end") 22 23#イメージマップ出力の段 24 25import matplotlib.pyplot as plt 26import matplotlib.gridspec as gridspec 27 28xsl = ags_data[::10] 29xmin=xsl.min() 30xmax=xsl.max() 31ymin=list1.start 32ymax=list1.stop 33 34fig =plt.figure(figsize=(8,4)) 35 36gs = gridspec.GridSpec(2,1) 37ax1 = fig.add_subplot(gs[0,:]) 38 39plt.plot(df.index,ags_data,linewidth=1) 40 41plt.title("sample_waveform") 42plt.xlabel("Time[s]") 43plt.ylabel("Amplitude") 44 45 46plt.grid(True) 47ax1.axis('tight') 48 49# plt.xlim([0,5]) 50 51ax2 = fig.add_subplot(gs[1,:]) 52 53im = ax2.imshow(corr_stack,extent=[xmin,xmax,ymin,ymax],interpolation='none',cmap='jet',origin='lower') 54 55plt.title("wavelet_coherence") 56plt.xlabel("Time[s]") 57plt.ylabel("Frequency[Hz]") 58plt.grid(linestyle='--', linewidth=0.5) 59plt.xticks() 60 61 62# カラーバーの追加をしたい ↓参考 63# https://matplotlib.org/2.0.2/examples/pylab_examples/colorbar_tick_labelling_demo.html 64 65cmlb=[0,0.01,0.02,0.03,0.04,0.05,0.06,0.07,0.08,0.09,0.1] 66cbar = fig.colorbar(im, ticks=cmlb, orientation='horizontal') 67cbar.ax.set_xticklabels(cmlb) # horizontal colorbar 68 69 70ax2.axis('tight')
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。