python wave文件的额处理方法

#!/usr/bin/env python
import numpy as np
import matplotlib.pyplot as plt
import scipy.signal as signal
import wave


'''
    This is a wave test program 
'''



def read_wave_data(file_path):  
    #open a wave file, and return a Wave_read object  
    f = wave.open(file_path,"rb")  
    #read the wave's format infomation,and return a tuple  
    params = f.getparams()  
    #get the info  
    nchannels, sampwidth, framerate, nframes = params[:4]  
    #Reads and returns nframes of audio, as a string of bytes.   
    str_data = f.readframes(nframes)  
    #close the stream  
    f.close()  
    #turn the wave's data to array  
    wave_data = np.fromstring(str_data, dtype = np.short)  
    #for the data is stereo,and format is LRLRLR...  
    #shape the array to n*2(-1 means fit the y coordinate)  
    wave_data.shape = -1, nchannels  
    #transpose the data  
    wave_data = wave_data.T  
    #calculate the time bar  
    time = np.arange(0, nframes) * (1.0/framerate)  
    return wave_data, nchannels,sampwidth,framerate 


def write_wave_data(file_path, wave_data, nchannels, sampwidth,framerate):
    f = wave.open(file_path,"wb")
    f.setnchannels(nchannels)
    f.setsampwidth(sampwidth)
    f.setframerate(framerate)
    f.writeframes(wave_data.tostring())
    return


if __name__ == "__main__":
    wave_data,nchannels,sampwidth,framerate = read_wave_data("voice_tv_door_binaural.wav")
    print nchannels,sampwidth,framerate
    print wave_data
    print type(wave_data)
    
print "OK"