代码改变世界

wav和PCM的转换(Python版本),pcm

2021-09-26 10:47  dreamboy2000  阅读(1258)  评论(0编辑  收藏  举报

wav和pcm是语音信号处理中最常见的文件格式,时常会遇到需要两种文件格式的相互转换。wav存储的一般是解码后为[-1, 1]的float数据,文件头有44个字节记录文件的采样率、长度等等信息。pcm存储的是int型整数,不含任何采样率相关信息。虽然原理比较简单,但是整理后更便于利用。

 

wav转pcm

原理:将文件头去掉,数据转成int型即可。

1 import numpy as np
2 def wav2pcm(wavfile, pcmfile, data_type=np.int16):
3     f = open(wavfile, "rb")
4     f.seek(0)
5     f.read(44)
6     data = np.fromfile(f, dtype= data_type)
7     data.tofile(pcmfile)

pcm转wav

原理:利用wave库,添加通道信息、采样位数、采样率等信息作为文件头,pcm数据直接写入即可。

 1 import wave
 2 def pcm2wav(pcm_file, wav_file, channels=1, bits=16, sample_rate=16000):
 3     pcmf = open(pcm_file, 'rb')
 4     pcmdata = pcmf.read()
 5     pcmf.close()
 6 
 7     if bits % 8 != 0:
 8         raise ValueError("bits % 8 must == 0. now bits:" + str(bits))
 9 
10     wavfile = wave.open(wav_file, 'wb')
11     wavfile.setnchannels(channels)
12     wavfile.setsampwidth(bits // 8)
13     wavfile.setframerate(sample_rate)
14     wavfile.writeframes(pcmdata)
15     wavfile.close()