scipy.io: не могу написать WAVFile

Question

У меня есть проблема, написание 2d Numpy массив в виде файла волны (аудио)

Согласно документу я должен написать 2d Int16 Numpy массива

https://docs.scipy.org/doc/scipy-0.18.1/reference/generated/scipy.io.wavfile.write.html

16-bit PCM -32768 +32767 int16 

Как мой NumPy массив в пределах (-1,1) в формате float32 я сначала преобразовать его в 16 бит Int

stereoAudio = ((stereoAudio * bits16max)).astype('int16') 

print "argmax : " + str(np.amax(stereoAudio)) 
print "argmin : " + str(np.amin(stereoAudio)) 

outWaveFileName = "out/file.wav" 
print "writing " + outWaveFileName 
wavfile.write(outWaveFileName,44100,stereoAudio) 

я получаю следующий результат:

argmax : 4389 
argmin : -4381 
writing out/file.wav 
Traceback (most recent call last): 
    File "/Users/me/file.py", line 132, in <module> 
wavfile.write(outWaveFileName,44100,stereoAudio) 
    File "//anaconda/lib/python2.7/site-packages/scipy/io/wavfile.py", line 353, in write 
    bytes_per_second, block_align, bit_depth) 
error: ushort format requires 0 <= number <= USHRT_MAX 

мои значений между -4391 и 4389 в 16bits форматирования должно быть в порядке. но мои данные интерпретируются как ushort

Answer 1

Функция write в scipy.io.wavfile ожидает, что входной массив будет иметь форму (num_samples, num_channels). Я подозреваю, что ваш массив имеет форму (num_channels, num_samples). Затем write пытается поставить num_samples в 16-битовое поле в структуре, которая записывается в WAV-файл, но значение num_samples слишком велико для 16-битного значения. (Обратите внимание, что если num_samples были достаточно малы, вы не получите сообщение об ошибке, но файл не будет иметь правильный формат.)

Быстрое исправление для записи транспонирования вашего массива:

wavfile.write(outWaveFileName, 44100, stereoAudio.T) 

Например, вот какой-то код, демонстрирующий ошибку; x и y имеют форму (2, 40000):

In [12]: x = (2*np.random.rand(2, 40000) - 1).astype(np.float32) 

In [13]: y = (x*32767).astype('int16') 

In [14]: from scipy.io import wavfile 

In [15]: wavfile.write('foo.wav', 44100, y) 
--------------------------------------------------------------------------- 
error          Traceback (most recent call last) 
<ipython-input-15-36b8cd0e729c> in <module>() 
----> 1 wavfile.write('foo.wav', 44100, y) 

/Users/warren/anaconda/lib/python2.7/site-packages/scipy/io/wavfile.pyc in write(filename, rate, data) 
    351 
    352   fmt_chunk_data = struct.pack('<HHIIHH', format_tag, channels, fs, 
--> 353          bytes_per_second, block_align, bit_depth) 
    354   if not (dkind == 'i' or dkind == 'u'): 
    355    # add cbSize field for non-PCM files 

error: ushort format requires 0 <= number <= USHRT_MAX 

Транспонирование массива так что входной wavfile.write имеет ожидаемую форму:

In [16]: wavfile.write('foo.wav', 44100, y.T) 

считаны данные, чтобы убедиться, что он работал, как ожидалось:

In [22]: fs, z = wavfile.read('foo.wav') 

In [23]: np.allclose(z, y.T) 
Out[23]: True