４章 q10 を変更

Author: 153hashimoto

khashimoto/chapter04/q10.py

@@ -5,15 +5,11 @@
 from q03 import stft
 
 
-def fr_convert(x, fs, S):
+def convert(x, fs, S):
     L = (len(x) - 1) * 2
     f = np.arange(len(x)) * fs / L
-    return f
-
-
-def t_convert(x, fs, S):
     t = np.arange(x.shape[1]) * S / fs
-    return t
+    return f, t
 
 
 # ４. の結果を再度プロット
@@ -30,8 +26,7 @@ def t_convert(x, fs, S):
 
 
 # プロット
-fr = fr_convert(x_stft.T, fs, S)
-t = t_convert(x_stft.T, fs, S)
+fr, t = convert(x_stft.T, fs, S)
 
 plt.subplot(1, 2, 1)
 plt.pcolormesh(t, fr, abs(x_stft).T)