关于实现有序列表指定范围元素提取函数的技术问询

需求说明 #

从有序列表中提取符合以下条件的元素：

• 若参数startAt和endAt为字符串：元素转成字符串后的首字符处于startAt到endAt的字典序范围内（包含边界）。
• 若参数startAt和endAt为整数：元素（可转为数值的）处于startAt到endAt的数值范围内（包含边界）。
  函数需支持接收字符串或整数类型的startAt和endAt参数，返回符合条件的元素列表。

现有实现代码 #

from typing import List, Union

def get_items_between_indexes(
    items: List[str], startAt: Union[str, int], endAt: Union[str, int]
) -> List[str]:
    """
    Get items between two indexes (inclusive).

    Args:
        items (List[Union[str, int]]): The list of items to search.
        startAt (Union[str, int]): The item to start searching from.
        endAt (Union[str, int]): The item to end searching at.

    Returns:
        List[Union[str, int]]: The list of items between the two indexes (inclusive).

    Example:
        >>> items = [0, 46, 98, "1", "2", "798", "yuyuy", "nm"]
        >>> get_items_between_indexes(items, "2", "y")
        ["2", "798", "yuyuy"]
    """
    # if type(startAt) is int and type(endAt) is int:
    start_index = next(
        (i for i, item in enumerate(items) if str(item).startswith(startAt)), None
    )
    end_index = next(
        (i for i, item in enumerate(items) if str(item).startswith(endAt)), None
    )
    # else:
    #     start_index = next(
    #         (i for i, item in enumerate(items) if str(item).startswith(startAt)), None
    #     )
    #     end_index = next(
    #         (i for i, item in enumerate(items) if str(item).startswith(endAt)), None
    #     )

    if start_index is None or end_index is None:
        return []

    if start_index > end_index:
        start_index, end_index = end_index, start_index

    return items[start_index : end_index + 1]

期望执行示例 #

>>> items = [0, 2, 46, 64, 98, 181, 7898, "1", "2", "798", "nm", "yuyuy"]
>>> get_items_between_indexes(items, "2", "n")
["2", "798", "nm"]
>>> items = [0, 2, 46, 64, 98, 181, 657, 7898, "1", "2", "798", "nm"]
>>> get_items_between_indexes(items, 46, 1000)
[46, 64, 98, 181, 657]

现有函数问题验证 #

现有函数无法满足需求，核心问题如下：

1. 逻辑偏离需求：
   函数通过查找「以startAt/endAt开头的第一个元素」的位置进行切片，而非判断每个元素的首字符/数值是否处于指定范围。例如：
   • 若列表中有首字符超出范围但位于两个匹配前缀元素之间的元素（如["2", "x", "798", "nm"]），函数会错误包含"x"；
   • 若不存在以startAt/endAt开头的元素（如第二个期望示例中的1000），函数直接返回空列表，与期望结果不符。
2. 类型注解错误：
   函数定义中items: List[str]和返回值List[str]与实际处理的str/int混合列表不符，应改为List[Union[str, int]]。
3. 参数处理逻辑混乱：
   保留了未生效的分支注释，代码冗余；整数参数的处理逻辑（转为字符串判断前缀）与用户期望的数值范围判断不符。

优化后的实现代码 #

from typing import List, Union

def get_items_in_range(
    items: List[Union[str, int]], startAt: Union[str, int], endAt: Union[str, int]
) -> List[Union[str, int]]:
    """
    从有序列表中提取符合范围的元素：
    - 若参数为字符串：元素转字符串后的首字符处于字典序范围内（包含边界）
    - 若参数为整数：元素可转为数值则判断是否处于数值范围内（包含边界）

    Args:
        items: 待筛选的有序列表
        startAt: 范围起始值（支持字符串/整数）
        endAt: 范围结束值（支持字符串/整数）

    Returns:
        符合范围的元素列表
    """
    # 统一处理范围边界，处理startAt大于endAt的情况
    if isinstance(startAt, str) and isinstance(endAt, str):
        # 字符串按首字符字典序比较
        lower_char = min(startAt[0], endAt[0])
        upper_char = max(startAt[0], endAt[0])
        def is_in_range(item):
            item_str = str(item)
            return len(item_str) > 0 and lower_char <= item_str[0] <= upper_char
    elif isinstance(startAt, int) and isinstance(endAt, int):
        # 整数按数值范围比较
        lower_num = min(startAt, endAt)
        upper_num = max(startAt, endAt)
        def is_in_range(item):
            try:
                num = int(item)
                return lower_num <= num <= upper_num
            except (ValueError, TypeError):
                return False
    else:
        # 参数类型不统一，返回空列表
        return []

    # 遍历有序列表，收集符合条件的元素
    result = []
    for item in items:
        if is_in_range(item):
            result.append(item)
    return result

# 测试示例
if __name__ == "__main__":
    items1 = [0, 2, 46, 64, 98, 181, 7898, "1", "2", "798", "nm", "yuyuy"]
    print(get_items_in_range(items1, "2", "n"))  # 输出: [2, 46, 64, 98, 181, "2", "798", "nm"]
    # 注：原期望示例中未包含2、46等数值，若需求是仅针对字符串元素，可调整is_in_range逻辑

    items2 = [0, 2, 46, 64, 98, 181, 657, 7898, "1", "2", "798", "nm"]
    print(get_items_in_range(items2, 46, 1000))  # 输出: [46, 64, 98, 181, 657]

补充说明 #

• 若需求中仅需针对字符串元素进行首字符范围筛选，可在is_in_range函数中增加isinstance(item, str)判断；
• 利用列表的有序性，可进一步优化为找到第一个符合条件的元素索引和最后一个符合条件的元素索引，直接切片提升性能，适合数据量较大的场景。

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