在音视频处理中，WAV是一种常见的音频封装格式，它包含音频数据的头部信息（如采样率、声道数、位深度等）以及实际的音频数据。使用Python为音频数据添加WAV格式的头部信息，可以通过`wave`库来实现。 ### 使用`wave`库创建WAV文件并添加头部信息 以下是一个完整的示例代码，展示如何使用`wave`模块创建WAV文件，并手动设置其头部信息： ```python import wave import numpy as np # 假设我们有以下音频参数 channels = 1 # 单声道 sample_width = 2 # 16位采样深度（2字节） frame_rate = 44100 # 采样率44.1kHz n_frames = 0 # 帧数（可设为0，后面写入时自动计算） comptype = "NONE" # 压缩类型（无压缩） compname = "not compressed" # 创建WAV文件对象 with wave.open("output.wav", 'wb') as wav_file: # 设置WAV文件的头部信息 wav_file.setparams((channels, sample_width, frame_rate, n_frames, comptype, compname)) # 生成一段示例音频数据（例如正弦波） duration = 5 # 秒 t = np.linspace(0, duration, int(frame_rate * duration), endpoint=False) audio_data = (32767 * np.sin(2 * np.pi * 440 * t)).astype(np.int16) # 440Hz 正弦波 # 将音频数据写入WAV文件 wav_file.writeframes(audio_data.tobytes()) ``` ### 手动构造WAV头信息 如果你已有原始音频数据（如PCM格式），并且希望手动构造WAV头部信息，可以使用`struct`模块构建WAV头。以下是一个示例： ```python import struct def write_wav_header(file, channels, sample_rate, bits_per_sample, data_size): byte_rate = sample_rate * channels * bits_per_sample // 8 block_align = channels * bits_per_sample // 8 bits_per_sample = bits_per_sample # RIFF header file.write(b'RIFF') file.write(struct.pack('<I', 36 + data_size)) # 文件总长度 - 8 file.write(b'WAVE') # fmt subchunk file.write(b'fmt ') file.write(struct.pack('<I', 16)) # fmt chunk size file.write(struct.pack('<H', 1)) # PCM format file.write(struct.pack('<H', channels)) file.write(struct.pack('<I', sample_rate)) file.write(struct.pack('<I', byte_rate)) file.write(struct.pack('<H', block_align)) file.write(struct.pack('<H', bits_per_sample)) # data subchunk file.write(b'data') file.write(struct.pack('<I', data_size)) ``` 使用该函数写入头部信息的完整示例： ```python # 假设我们有原始PCM音频数据 pcm_data = np.random.randint(-32768, 32767, size=44100 * 5, dtype=np.int16) with open("output_with_header.wav", "wb") as f: write_wav_header(f, channels=1, sample_rate=44100, bits_per_sample=16, data_size=len(pcm_data) * 2) f.write(pcm_data.tobytes()) ``` 上述代码通过手动构造WAV文件的RIFF头和fmt子块，确保音频数据可以被标准播放器识别[^3]。 ### 注意事项 - `wave`模块仅支持单声道或双声道的WAV文件，不支持多声道音频。 - 若需要处理更复杂的音频格式（如MP3、OGG），建议使用`pydub`结合`ffmpeg`[^2]。 - 确保音频数据的字节顺序和格式与WAV头部定义的参数一致，否则可能导致播放失败。 ---