如何高效获取Python列表中非None值的连续索引区间？

我来帮你搞定这个问题！你的需求是找出列表中非None值所在的连续索引区间，先聊聊你现有代码的问题，再给你两个简洁可靠的实现方案。

现有代码的问题分析 #

你的代码逻辑绕了太多弯，比如在处理元素为None时错误地将起始索引s设为i+1（但i+1可能还是None），结尾的判断逻辑也混乱，导致出现[6,5]这种反向区间，以及把None元素的索引错误包含进去的情况。咱们换个更清晰的思路来实现。

方案一：直观遍历追踪法 #

这个方法逻辑简单易懂，通过追踪连续非None区间的起始索引，遇到None时结束区间，遍历结束后处理最后一个可能未闭合的区间：

def get_not_None_ranges(list_):
    result = []
    start = None
    for idx, elem in enumerate(list_):
        if elem is not None:
            if start is None:
                # 开始一个新的非None区间
                start = idx
        else:
            if start is not None:
                # 结束当前非None区间
                result.append([start, idx - 1])
                start = None
    # 处理列表末尾的非None区间
    if start is not None:
        result.append([start, len(list_) - 1])
    return result

测试你的案例：

test1 = [None, 0, None]
test2 = [2,1,None]
test3 = [None,None,3]
test4 = [1,0,None,0,0,None,None,1,None,0]

print(get_not_None_ranges(test1))  # [[1,1]] 符合预期
print(get_not_None_ranges(test2))  # [[0,1]] 符合预期
print(get_not_None_ranges(test3))  # [[2,2]] 符合预期
print(get_not_None_ranges(test4))  # [[0,1],[3,4],[7,7],[9,9]] 符合预期

方案二：Pythonic的itertools.groupby法 #

利用itertools.groupby可以按元素是否为None分组，然后直接计算每组的索引区间，代码更简洁：

import itertools

def get_not_None_ranges(list_):
    result = []
    current_idx = 0
    for is_none, group in itertools.groupby(list_, key=lambda x: x is None):
        group_length = len(list(group))
        if not is_none:
            # 非None组，计算起始和结束索引
            start = current_idx
            end = current_idx + group_length - 1
            result.append([start, end])
        current_idx += group_length
    return result

这个方法同样能通过所有测试案例，而且代码更紧凑，适合熟悉Python标准库的开发者使用。

两种方案的对比 #

• 遍历追踪法：逻辑直观，容易理解和调试，适合新手。
• groupby法：代码简洁优雅，利用Python标准库减少重复代码，适合追求代码简洁性的场景。

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