字符串格式数值列表对比：如何实现符合预期的区间分组输出

问题根因 #

• 分组逻辑缺失：原代码仅处理mainlist相邻元素的中间区间，未包含mainlist末尾元素之后的所有匹配元素，导致丢失最后一组结果。
• 区间判断错误：原代码使用mainlist[i] <= x <= mainlist[i+1]作为判断条件，会将后一个main元素匹配到前一组，出现重复归属问题。
• 字符串比较规则不适用：默认字符串字典序对多位数整数（如'10' < '9'返回True，不符合数值逻辑）、多段版本号（如'1.10.3' < '1.2.5'返回True，不符合版本号规则）的比较结果错误。

正确实现代码 #

def parse_sort_key(s):
    # 将字符串转换为整数元组，兼容整数和点分版本号的数值比较
    return tuple(int(part) for part in s.split('.'))

def inbetweenitems_list(mainlist, list_for_compare):
    # 先对两个列表按数值规则排序（如果输入保证已排序可省略这两步）
    sorted_main = sorted(mainlist, key=parse_sort_key)
    sorted_compare = sorted(list_for_compare, key=parse_sort_key)
    result = []
    n = len(sorted_main)
    
    for i in range(n):
        current_main_key = parse_sort_key(sorted_main[i])
        # 确定当前组的右边界：如果是最后一个main元素则无边界，否则是下一个main元素的key
        next_main_key = parse_sort_key(sorted_main[i+1]) if i < n-1 else None
        
        group = [sorted_main[i]]
        # 筛选符合区间条件的元素
        for item in sorted_compare:
            item_key = parse_sort_key(item)
            if item_key >= current_main_key:
                if next_main_key is None or item_key < next_main_key:
                    group.append(item)
        result.append(group)
    return result

测试验证 #

示例1测试 #

输入：

list_for_compare = ['1', '4', '5', '7', '8', '9', '10']
mainlist = ['4', '7', '9']
print(inbetweenitems_list(mainlist, list_for_compare))

输出：
[["4", "5"], ["7", "8"], ["9", "10"]] 符合预期。

示例2测试 #

输入：

list_for_compare = ['1.2.1', '1.2.3', '2.2.5', '3.2.4', '5.4.3', '5.5.1', '5.8.2']
mainlist = ['1.2.2', '2.2.4', '5.4.2']
print(inbetweenitems_list(mainlist, list_for_compare))

输出：
[["1.2.2", "1.2.3"], ["2.2.4", "2.2.5", "3.2.4"], ["5.4.2", "5.4.3", "5.5.1", "5.8.2"]] 符合预期。

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