如何从递增浮点数列表中提取各整数首次出现的对应元素？

问题背景 #

你需要从递增的浮点数列表里，找到指定目标整数（比如1、2、3）对应的第一个符合条件的元素——示例里1对应第一个>=1且<2的1.4，2对应第一个>=2且<3的2.1，3对应第一个>=3的3，最后把这些元素存入新列表。

你的现有代码能得到预期结果，但存在几个可以优化的点：

• 硬编码了目标整数和结果列表长度，没法灵活适配其他目标（比如要找4、5）
• 用0作为初始值，如果列表里有0元素会导致误判
• 没有充分利用列表递增的特性，做了一些不必要的判断

优化实现 #

方案1：一次遍历法（O(n)时间） #

利用列表递增的特性，遍历一次就能完成所有目标的查找，找到所有目标后直接终止遍历，减少不必要的循环：

def get_first_matches(float_list, target_integers):
    result = []
    target_pos = 0
    total_targets = len(target_integers)
    
    for num in float_list:
        if target_pos >= total_targets:
            break  # 所有目标都找到了，不用继续遍历
        
        current_target = target_integers[target_pos]
        # 判断当前数是否符合目标要求：
        # 非最后一个目标：找第一个 >=目标 且 <目标+1 的数
        # 最后一个目标：找第一个 >=目标 的数
        if num >= current_target:
            if target_pos < total_targets - 1:
                if num < current_target + 1:
                    result.append(num)
                    target_pos += 1
            else:
                result.append(num)
                target_pos += 1
    # 处理列表中没有对应元素的目标（按需调整默认值）
    while target_pos < total_targets:
        result.append(None)
        target_pos += 1
    return result

# 测试示例
float_list = [0.3, 0.8, 1.4, 1.9, 2.1, 2.7, 3, 3.2, 3.8]
targets = [1, 2, 3]
print(get_first_matches(float_list, targets))  # 输出: [1.4, 2.1, 3]

这个方案的好处是：

• 只遍历一次列表，找到所有目标后提前结束，效率高
• 支持任意数量的目标整数，扩展性强
• 用None作为未找到的默认值，避免和列表元素冲突

方案2：二分查找法（O(logn)时间） #

因为列表是递增的，用二分查找可以快速定位到第一个符合条件的元素，适合处理大规模列表：

import bisect

def find_first_for_target(float_list, target, is_last_target):
    # 找到第一个 >= target 的元素索引
    idx = bisect.bisect_left(float_list, target)
    if idx >= len(float_list):
        return None
    
    num = float_list[idx]
    # 根据是否是最后一个目标判断条件
    if not is_last_target:
        return num if num < target + 1 else None
    else:
        return num

def get_first_matches(float_list, target_integers):
    result = []
    for i, target in enumerate(target_integers):
        is_last = (i == len(target_integers) - 1)
        result.append(find_first_for_target(float_list, target, is_last))
    return result

# 测试示例
float_list = [0.3, 0.8, 1.4, 1.9, 2.1, 2.7, 3, 3.2, 3.8]
targets = [1, 2, 3]
print(get_first_matches(float_list, targets))  # 输出: [1.4, 2.1, 3]

这个方案的优势是：

• 每个目标的查找时间是O(logn)，当列表很大、目标数量多的时候，比一次遍历效率高很多
• 用Python内置的bisect模块，代码简洁且经过验证，可靠性高

对原有代码的改进建议 #

1. 把目标整数做成参数传入，不要硬写在代码里，方便后续修改
2. 别用0当初始值，换成None这种不会和列表元素混淆的默认值
3. 找到所有目标后直接break，不用再遍历剩下的元素

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