Python 和音 生成した音のスペクトログラム表示をさせたいです。

Question

前提・実現したいこと

write('hogehoge.wav', Fs, np.int16(s.getSound()).reshape((N, 1)))までの部分を実行するとドミソの和音が生成されます。その上で、その音の波形をスペクトログラム表示させたいのですが、以下のようなエラーが出てしまいます。どのようにしたら解決できるか教えてください。よろしくお願いします。

発生している問題・エラーメッセージ

Traceback (most recent call last):
  File "ans2.py", line 46, in <module>
    Sxx, f, t, im = plt.specgram(np.int16(s.getSound()).reshape((N, 1)), Fs = int(Fs), cmap='jet')
  File "/usr/local/anaconda3/lib/python3.6/site-packages/matplotlib/pyplot.py", line 3448, in specgram
    vmin=vmin, vmax=vmax, data=data, **kwargs)
  File "/usr/local/anaconda3/lib/python3.6/site-packages/matplotlib/__init__.py", line 1717, in inner
    return func(ax, *args, **kwargs)
  File "/usr/local/anaconda3/lib/python3.6/site-packages/matplotlib/axes/_axes.py", line 7170, in specgram
    mode=mode)
  File "/usr/local/anaconda3/lib/python3.6/site-packages/matplotlib/mlab.py", line 1288, in specgram
    mode=mode)
  File "/usr/local/anaconda3/lib/python3.6/site-packages/matplotlib/mlab.py", line 713, in _spectral_helper
    result = stride_windows(x, NFFT, noverlap, axis=0)
  File "/usr/local/anaconda3/lib/python3.6/site-packages/matplotlib/mlab.py", line 545, in stride_windows
    raise ValueError('only 1-dimensional arrays can be used')
ValueError: only 1-dimensional arrays can be used

該当のソースコード

# -*- coding: utf-8 -*-

import numpy as np
from scipy.io.wavfile import write
import matplotlib.pyplot as plt


class GenFreq(object):
    def __init__(self, Fs=44100., dur=1.):
        self.Fcs = {'C':262., 'D':294., 'E':330., 'F':349., 'G':392., 'A':440., 'B':494.}
        self.Amp = 1.0
        self.Fs = Fs
        self.dur = dur
        self.Num = int( dur * Fs )
        self.x = np.zeros( self.Num )
        self.m = np.zeros( self.Num )
        delta = 1./Fs
        self.t = np.arange( self.Num ) * delta

    def getSound(self):
        return self.x

    def getLen(self):
        return self.Num

    def addTone(self, code='C'):
        f = self.Fcs[code]
        self.m = self.Amp * np.sin( 2. * np.pi * f * self.t )
        self.x = self.x + self.m

#           
#         (CEG)       

Fs = 44100
s = GenFreq(Fs, dur=1.)
s.addTone('C')
s.addTone('E')
s.addTone('G')
N = s.getLen()

# 1. scipy.wav.iofile   write                    
# 2. scipy.signal.spectrogram      matplotlib.pylab.specgram                    
write('hogehoge.wav', Fs, np.int16(s.getSound()).reshape((N, 1)))

fig = plt.figure(figsize=(8,6))
Sxx, f, t, im = plt.specgram(np.int16(s.getSound()).reshape((N, 1)), Fs = int(Fs), cmap='jet')
plt.pcolormesh(t, f, np.log(Sxx), cmap='jet')
plt.xlim(0, 2.0e5)
plt.show()

試したこと

write('hogehoge.wav', Fs, np.int16(s.getSound()).reshape((N, 1)))までを記述するとドミソの和音が正しく生成されます。しかし、fig = plt.figure(figsize=(8,6))以下を書くとエラーが出てしまいました。

補足情報（FW/ツールのバージョンなど）

Linux環境です。Python3を使用しています。

Comments

[8524ba23]

参考にしたURLがあれば記載ください。

[Teru_miyanaga]

https://teratail.com/questions/278801
こちらの回答者様が答えられていたコードを元に書きました。

Answer 1

単に表示させたいだけなら以下でよいのではないでしょうか。

fig = plt.figure(figsize=(8,6))
#Sxx, f, t, im = plt.specgram(np.int16(s.getSound()).reshape((N, 1)), Fs = int(Fs), cmap='jet')
Sxx, f, t, im = plt.specgram(np.int16(s.getSound()), Fs = int(Fs), cmap='jet')
#plt.pcolormesh(t, f, np.log(Sxx), cmap='jet')
#plt.xlim(0, 2.0e5)
plt.show()

Comments

[Teru_miyanaga]

解決できました。
ありがとうございました。