频域波束成形声源定位代码

以下是一个简单的频域波束成形声源定位的示例代码：

import numpy as np
from scipy.signal import stft

def beamforming_localization(signal, fs, angles):
    # 进行STFT转换
    _, _, stft_data = stft(signal, fs=fs)
    
    # 计算波束权重
    num_mics = stft_data.shape[1]
    num_angles = len(angles)
    weights = np.zeros((num_mics, num_angles), dtype=np.complex128)
    
    for i in range(num_angles):
        angle = angles[i]
        for mic in range(num_mics):
            phase_shift = mic * np.cos(angle) * 2 * np.pi / fs
            weights[mic, i] = np.exp(-1j * phase_shift)
    
    # 进行波束成形
    beamformed_data = np.dot(stft_data, weights)
    
    # 取波束成形结果的幅度谱
    beamformed_mag = np.abs(beamformed_data)
    
    # 根据幅度谱进行声源定位
    source_location = np.argmax(beamformed_mag, axis=1)
    
    return source_location

# 示例用法
# 假设有4个麦克风
num_mics = 4

# 假设采样率为44100Hz
fs = 44100

# 假设待处理的音频信号为signal，长度为N
signal = np.random.randn(N, num_mics)

# 假设我希望在-90度到90度范围内进行声源定位，每隔1度采样一个角度
angles = np.arange(-90, 91, 1)

# 进行声源定位
source_location = beamforming_localization(signal, fs, angles)

请注意，以上代码仅为示例，实际应用中可能需要根据具体情况进行适当的修改和优化。此外，还需要注意音频信号的预处理、噪声抑制等问题，以获得更准确的声源定位结果。