如何按前缀分组WAV音频文件？List/Tuple选型及优化方案咨询

一、文件分组实现 #

可以用字典按前缀（如ZOOM0001）分组，键为前缀，值为对应组的文件路径列表，代码实现如下：

import glob
import os

# 目标文件夹路径
target_dir = r'C:\Users\adam\Music\temp_h6'
# 获取所有WAV文件路径
files = glob.glob(os.path.join(target_dir, '*.wav'))

# 初始化分组字典
grouped_files = {}
for file_path in files:
    # 提取文件名（不含路径和后缀）
    filename = os.path.splitext(os.path.basename(file_path))[0]
    # 拆分前缀（如ZOOM0001）和后缀（如LR）
    prefix = filename.split('_')[0]
    # 将文件添加到对应分组
    if prefix not in grouped_files:
        grouped_files[prefix] = []
    grouped_files[prefix].append(file_path)

# 若需要将分组转为列表的列表格式
grouped_list = list(grouped_files.values())

二、List vs Tuple选择 #

优先用List存储分组结果，原因：

• 你需要单独处理组内每个文件，后续可能需要调整文件顺序、临时修改组内容（比如移除损坏文件），List是可变类型，操作更灵活；
• Tuple是不可变类型，创建后无法修改，不利于后续处理流程的调整。

三、分组混音实现 #

以pydub库为例（需先安装：pip install pydub，并确保系统安装了ffmpeg），实现每组音频的混音处理：

1. 混音函数 #

from pydub import AudioSegment

def mix_group_audio(file_group, output_dir):
    # 加载组内第一个音频作为基础轨道
    mixed_track = AudioSegment.from_wav(file_group[0])
    # 依次叠加其他音频轨道
    for audio_path in file_group[1:]:
        current_track = AudioSegment.from_wav(audio_path)
        # 统一音频长度（可选，可根据需求调整为补全或截断）
        min_duration = min(len(mixed_track), len(current_track))
        mixed_track = mixed_track[:min_duration]
        current_track = current_track[:min_duration]
        # 叠加音频（默认混合模式，可通过参数调整音量、位置等）
        mixed_track = mixed_track.overlay(current_track)
    
    # 生成输出文件名并导出
    prefix = os.path.basename(file_group[0]).split('_')[0]
    output_path = os.path.join(output_dir, f"{prefix}_MIXED.WAV")
    mixed_track.export(output_path, format="wav")

2. 批量处理所有分组 #

# 创建输出目录（不存在则自动创建）
output_dir = os.path.join(target_dir, "mixed_output")
os.makedirs(output_dir, exist_ok=True)

# 遍历每个分组进行混音
for group in grouped_files.values():
    # 可选：对组内文件按固定顺序排序（确保LR、TR1、TR3的处理顺序）
    def sort_audio_files(file_path):
        suffix = os.path.splitext(os.path.basename(file_path))[0].split('_')[1]
        # 定义优先级：LR > TR1 > TR3
        priority = {"LR": 0, "TR1": 1, "TR3": 2}
        return priority.get(suffix, 3)
    
    group.sort(key=sort_audio_files)
    # 执行混音
    mix_group_audio(group, output_dir)

四、优化建议 #

• 错误处理：添加try-except块捕获文件加载失败、格式不支持等异常，避免批量处理中断；
• 性能优化：若处理大量音频，可采用多线程/多进程并行处理（注意控制并发数，避免CPU/IO过载）；
• 自定义混音规则：可根据需求调整overlay方法的参数（如gain_during_overlay调整叠加音量、position设置轨道偏移）。

内容的提问来源于stack exchange，提问作者ben