【Python】初心者、フーリエ変換、IndexError: list index out of rangeの解決法

Question

ステレオのwavファイルに対し、フーリエ変換を行いたいので、以下のサイトを参考にし、実行しました。
参考コード

IndexError: list index out of rangeというエラーが吐き出されました。

原因につきまして、存在しない要素にアクセスしようとしている、ということはわかったのですが、

具体的にどこの箇所が原因源に該当するのか、どこを直せばいいのかなど不明ですので、よろしくお願いします。

また、実行に使用した音源は以下です。
使用ステレオ音源

import wave
import struct
from scipy import fromstring, int16
import numpy as np
from pylab import *
%matplotlib inline

wavfile = 'ASMR_grain/bonfire_out.wav'
wr = wave.open(wavfile, "rb")
ch = wr.getnchannels()
width = wr.getsampwidth()
fr = wr.getframerate()
fn = wr.getnframes()

N = 64
span = 4

print('チャンネル', ch)
print('総フレーム数', fn)
print('サンプル時間', 1.0 * N * span / fr, '秒')

origin = wr.readframes(wr.getnframes())
data = origin[:N * span * ch * width]
wr.close()

print('現配列長', len(origin))
print('サンプル配列長: ', len(data))

チャンネル 2
総フレーム数 1434
サンプル時間 0.005804988662131519 秒
現配列長 5736
サンプル配列長: 1024

# ステレオ前提
X = np.frombuffer(data, dtype="int16")
left = X[::2]
right = X[1::2]

def fourier (x, n, w):
    K = []
    for i in range(0, w-2):
        sample = x[i * n:( i + 1) * n]
        partial = np.fft.fft(sample)
        K.append(partial)

    return K

def inverse_fourier (k):
    ret = []
    for sample in k:
        inv = np.fft.ifft(sample)
        ret.extend(inv.real)

    print (len(sample))
    return ret

Kl = fourier(left, N, span)
Kr = fourier(right, N, span)
freqlist = np.fft.fftfreq(N, d=1/fr)
amp = [np.sqrt(c.real ** 2 + c.imag ** 2) for c in Kl[2500]]
plot(freqlist, amp, marker= 'o', linestyle='-')
axis([0, fr / 2 , 0, 100000])

amp = [np.sqrt(c.real ** 2 + c.imag ** 2) for c in Kr[2500]]
plot(freqlist, amp, marker= 'o', linestyle='-')

エラー

IndexError Traceback (most recent call last)
<ipython-input-33-ee594416e8d3> in <module>
2 Kr = fourier(right, N, span)
3 freqlist = np.fft.fftfreq(N, d=1/fr)
----> 4 amp = [np.sqrt(c.real ** 2 + c.imag ** 2) for c in Kl[2]]
5 plot(freqlist, amp, marker= 'o', linestyle='-')
6 axis([0, fr / 2 , 0, 10])

IndexError: list index out of range

Comments

[coco_bauer]

「amp = [np.sqrt(c.real ** 2 + c.imag ** 2) for c in Kl[2500]]」の行で、
「----> 4 amp = [np.sqrt(c.real 2 + c.imag 2) for c in Kl[2]]」というエラーが出ているように見えるのですが、"c.real ** 2 "が”c.real 2 ”に書き換わってたりして意味不明です。
本当に質問に書かれているPythonのコードを実行した際にでたエラーなのですか？？？

[Kenza]

すいません、表記ミスです

Kl = fourier(left, N, span)
Kr = fourier(right, N, span)
freqlist = np.fft.fftfreq(N, d=1/fr)
amp = [np.sqrt(c.real ** 2 + c.imag ** 2) for c in Kl[2]]
plot(freqlist, amp, marker= 'o', linestyle='-')
axis([0, fr / 2 , 0, 100000])

amp = [np.sqrt(c.real ** 2 + c.imag ** 2) for c in Kr[2]]
plot(freqlist, amp, marker= 'o', linestyle='-')

が実際に実行したコードです

Answer 1

amp = [np.sqrt(c.real ** 2 + c.imag ** 2) for c in Kl[2]]

こちらの行で、Kl[2]は存在しません。KlとKrの長さはそれあぞれ2なので、参照できるインデックスは0か1です(つまりエラーにならないのはKl[0]かKl[1]などと書いた場合のみ)
試しにこちらを

amp = [np.sqrt(c.real ** 2 + c.imag ** 2) for c in Kl[1]]

amp = [np.sqrt(c.real ** 2 + c.imag ** 2) for c in Kr[2]]

と書き換えたところ、動きました。作者さんの意図と合致するかは別問題ですが、、

Comments

[Kenza]

ありがとうございます。解決しました！