如何修复Python提取子列表首个元素函数的空列表报错问题

原函数遇到空列表时会触发IndexError，因为空列表不存在[0]索引对应的元素。以下是几种实用的修复方案：

方案1：过滤空列表 #

直接跳过空的子列表，只返回非空子列表的首个元素：

def firstelements(w):
    return [item[0] for item in w if item]

调用firstelements([[10,10],[3,5],[]])会得到结果[10, 3]。

方案2：为空列表设置默认值 #

如果需要保证结果列表长度和输入一致，可以给空列表对应的位置指定默认值（比如None）：

def firstelements(w):
    return [item[0] if item else None for item in w]

调用后会返回[10, 3, None]。

方案3：异常捕获处理 #

用try-except捕获可能的索引错误或类型错误，兼容更多异常场景：

def firstelements(w):
    result = []
    for item in w:
        try:
            result.append(item[0])
        except (IndexError, TypeError):
            result.append(None)  # 可替换为你需要的其他默认值
    return result

内容的提问来源于stack exchange，提问作者Gonçalo Barreto