如何使用Python中的Soundfile将具有指定SNR的高斯噪声添加到音频文件中？

pip install soundfile

import soundfile as sf
import numpy as np
import random

def generate_gaussian_noise(signal, snr):
    # 计算信号的能量
    signal_power = np.sum(signal ** 2) / len(signal)
    
    # 根据信噪比计算噪声的能量
    noise_power = signal_power / (10 ** (snr / 10))
    
    # 生成与信号长度相同的高斯噪声
    noise = np.random.normal(0, np.sqrt(noise_power), len(signal))
    
    return noise

filename = "path/to/audio/file.wav"
audio_data, sample_rate = sf.read(filename)

snr = 20  # 以dB为单位的信噪比
noise = generate_gaussian_noise(audio_data, snr)

noisy_audio = audio_data + noise

output_filename = "path/to/output/file.wav"
sf.write(output_filename, noisy_audio, sample_rate)