NumPy FFT元组索引超出范围?

时间:2015-06-24 17:05:21

标签: python numpy fft

我上周运行ffts没问题,现在我遇到了一些奇怪的错误:

Traceback (most recent call last):
  File "call.py", line 52, in <module>
    main()
  File "call.py", line 50, in main
    validate(temp_path)
  File "call.py", line 35, in validate
    ivr_wav = IVR_Wav(validate_path)
  File "call.py", line 42, in __init__
    self.fft = get_fft(temp_wav_path)
  File "call.py", line 21, in get_fft
    return numpy.fft.fft(frames)
  File "/usr/lib/python2.7/dist-packages/numpy/fft/fftpack.py", line 174, in fft
    return _raw_fft(a, n, axis, fftpack.cffti, fftpack.cfftf, _fft_cache)
  File "/usr/lib/python2.7/dist-packages/numpy/fft/fftpack.py", line 50, in _raw_fft
    n = a.shape[axis]
IndexError: tuple index out of range

我的代码是:

def get_fft(fft_path):
    wav = wave.open(fft_path, 'r')
    frames = wav.readframes(wav.getnframes())
    wav.close()
    return numpy.fft.fft(frames)

和NumPy代码是:

def _raw_fft(a, n=None, axis=-1, init_function=fftpack.cffti,work_function=fftpack.cfftf, fft_cache = _fft_cache ):
    a = asarray(a)
    if n is None:
         n = a.shape[axis]
    if n < 1:
         raise ValueError("Invalid number of FFT data points (%d) specified." % n)

-1超出元组的范围是有道理的,但我该如何解决这个问题呢?我不太清楚,但我相信调用numpy.fft.fft(path, axis=0)应该允许n=a.shape[axis]执行,但这会将n设置为样本中的长度吗?这甚至是正确的吗?我对numpy和nd-arrays并不是很熟悉。

1 个答案:

答案 0 :(得分:1)

问题不是你的FFT,而是readFrames命令。 wav.readFrames返回Bytes作为输出,它们肯定不是Numpy FFT所需的类似数组。

看看这里:

What is returned by wave.readframes?

我怀疑你认为这会像MATLAB waveRead命令一样,它返回一个浮点数组,但这似乎不是这里的情况。

编辑:我使用map()命令使其工作。

waveform = map(ord, list(frames))
plt.plot(np.fft.fft(waveform))
plt.show()