Python实现按文件名分组并返回含最大ID的文件的问题

云上老周

2026-8-16

按文件前缀分组并筛选最大ID文件的解决方案

你的代码问题在于字典键值逻辑错误，把ID作为键会导致同前缀文件被覆盖，无法实现分组比较。下面是两种可行的实现方法：

方法一：单次遍历实时更新最大值

这种方法仅遍历一次目录，实时维护每个前缀组的最大ID文件，效率较高：

import os

directory = '/directory'
grouped_files = {}

for file in os.listdir(directory):
    # 跳过不符合格式的文件
    if '_' not in file or not file.endswith('.txt'):
        continue
    # 只分割一次下划线，避免前缀含下划线时出错
    prefix, rest = file.split('_', 1)
    id_str = rest.split('.')[0]
    # 转ID为整数，避免字符串比较的逻辑误差
    try:
        file_id = int(id_str)
    except ValueError:
        continue
    
    # 更新当前前缀组的最大ID文件
    if prefix not in grouped_files or file_id > grouped_files[prefix][0]:
        grouped_files[prefix] = (file_id, file)

# 提取最终结果
result = [file for _, file in grouped_files.values()]
print(result)

方法二：先分组再取最大值

这种方法先用字典收集同前缀的所有文件，再对每个分组筛选最大值，逻辑更直观：

import os
from collections import defaultdict

directory = '/directory'
groups = defaultdict(list)

for file in os.listdir(directory):
    if '_' not in file or not file.endswith('.txt'):
        continue
    prefix, rest = file.split('_', 1)
    id_str = rest.split('.')[0]
    try:
        file_id = int(id_str)
    except ValueError:
        continue
    # 将当前文件按前缀分组存入列表
    groups[prefix].append((file_id, file))

# 对每个分组取ID最大的文件
result = [max(items, key=lambda x: x[0])[1] for items in groups.values()]
print(result)