pythonでx=aの時のyの値を出力したい

pythonで、スペクトルの値(y軸)が最大値をとるときの周波数(x軸)を出力したいと考えています。
以下のプログラムで、それぞれのkについてスペクトルの最大値を出力することには成功しました。しかし、その時の周波数を求めるために書いた最後の行は意味をなしていないように思えます。また、mAdftが一回ごとに書き換えられてしまっており、配列の形になっていないと思っています。

#cording:utf-8
import numpy as np
import pylab
import wave

def wavread(filename):
    wf = wave.open(filename, "r")
    fs = wf.getframerate()
    x = wf.readframes(wf.getnframes())
    x = np.frombuffer(x, dtype="int16") / 32768.0  # (-1, 1)に正規化
    wf.close()
    return x, float(fs)

if __name__ == "__main__":
    # 波形を表示
    wav, fs = wavread("eiyuu.wav")
    t = np.arange(0.0, len(wav) / fs, 1/fs)
    pylab.plot(t * 1000, wav)
    pylab.xlabel("time [ms]")
    pylab.ylabel("amplitude")
    #pylab.show()

    # 母音の定常部分（中心部）のスペクトルを
    center = int(len(wav) / 2)  # 中心のサンプル番号
    cuttime = len(wav)/12  # 切り出す長さ [s]

    for k in range(12):
        wavdata = wav[(k*int(cuttime)): ((k+1)*int(cuttime))]
        time = t[(k*int(cuttime)): ((k+1)*int(cuttime))]
        pylab.subplot(211)
        pylab.plot(time * 1000, wavdata)
        pylab.ylabel("amplitude")

    # ハニング窓をかける
        hanningWindow = np.hanning(len(wavdata))
        wavdata = wavdata * hanningWindow

        pylab.subplot(212)
        pylab.plot(time * 1000, wavdata)
        pylab.xlabel("time [ms]")
        pylab.ylabel("amplitude")
        pylab.show()


    # 切り出した音声のスペクトルを求める
        n = 2048  # FFTのサンプル数
    # 離散フーリエ変換
        dft = np.fft.fft(wavdata, n)
    # 振幅スペクトル
        Adft = np.abs(dft)
    # パワースペクトル
        Pdft = np.abs(dft) ** 2
    # 周波数スケール
        fscale = np.fft.fftfreq(n, d=1.0 / fs)
    # プロット
        pylab.subplot(211)
        pylab.plot(fscale[0:int(n / 2)], Adft[0:int(n / 2)])
        pylab.xlabel("frequency [Hz]")
        pylab.ylabel("amplitude spectrum")
        pylab.xlim(0, 1000)

        pylab.subplot(212)
        pylab.plot(fscale[0:int(n / 2)], Pdft[0:int(n / 2)])
        pylab.xlabel("frequency [Hz]")
        pylab.ylabel("power spectrum")
        pylab.xlim(0, 1000)

        pylab.show()
        mAdft=[max(Adft)]
        print(mAdft)
        print(fscale)

そこで、プログラム最終部分をif文を使って以下のように書き換えたところ、後述するようなエラーが起こってしまいました。

    mAdft=[max(Adft)]
    print(mAdft)
    print(fscale)
    if Adft==mAdft:
        print(fscale)

if Adft==mAdft:
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

おそらく、Adft,mAdft,fscaleがすべて配列であることが原因のエラーのようですが、解決策までは掴めませんでした。
繰り返しになりますが、for文のそれぞれの出力結果をすべて配列に入れる方法、および、スペクトルが最大値の時の周波数の値を出力する方法を教えていただきたいです。

行動規範の内容に同意します

回答1件

ベストアンサー

AdftとmAdftの要素数合ってますか？？

forループの中で毎回
mAdft=[max(Adft)]
を呼び出した場合、mAdftはkが最後の時のmax(Adft)が１つだけ入った配列になるかと思います。
ループの前で空のmAdftを定義してあげてから
np.append(mAdft,max(Adft))
としないといけないような気がします。

間違ってたらすみません

投稿2020/01/11 16:59