Python如何在嵌套列表中查找指定元素并返回对应子列表

这个需求完全可以实现，核心逻辑非常简单：遍历外层嵌套列表的每一个子列表，判断目标元素是否存在于当前子列表中，命中时直接返回对应子列表即可。

基础实现（返回第一个匹配的子列表） #

对应你给出的两层嵌套结构，最简洁的实现如下：

def find_target_sublist(nested_lists, search_value):
    for sublist in nested_lists:
        if search_value in sublist:
            return sublist
    # 未找到匹配项时可根据业务需求自定义返回值，比如返回None或者抛出异常
    return None

# 调用测试
lists = [ [1,2,3] , [4,5,6] , [8,9,10] ]
result = find_target_sublist(lists, 5)
print(result)  # 输出结果：[4,5,6]

多匹配场景实现（返回所有包含目标元素的子列表） #

如果你的数据里可能有多个子列表同时包含搜索值，需要返回全部匹配结果，可以用列表推导式直接收集：

def find_all_matched_sublists(nested_lists, search_value):
    return [sublist for sublist in nested_lists if search_value in sublist]

补充说明 #

• 你之前看到的仅返回布尔值的搜索方法，本质是只完成了存在性校验，只要在校验命中时不返回True，而是返回当前遍历到的子列表对象，就能拿到完整的子列表内容
• 上述实现针对两层嵌套结构编写，如果你的列表有更深的嵌套层级，可以在遍历逻辑里加递归判断即可适配
• 常规业务场景下这个实现的性能完全够用，如果是超大数据量的高频查询场景，可以提前构建「元素-所属子列表」的映射字典，把单次查询的时间复杂度降到O(1)

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