如何遍历嵌套列表并查找目标元素所在子列表的索引

需求说明 #

遍历指定嵌套列表，搜索目标数字，返回该数字所在子列表的索引。给定示例嵌套列表：

my_list = [['Hello', 'World','!'], [1, 3, 'Test', 5, 7], [2, 4]]

比如搜索数字1时返回1，搜索2时返回2。

实现代码 #

def find_sublist_index(nested_list, target_num):
    for idx, sublist in enumerate(nested_list):
        if target_num in sublist:
            return idx
    # 未找到目标数字时返回None
    return None

使用示例 #

# 定义示例嵌套列表
my_list = [['Hello', 'World','!'], [1, 3, 'Test', 5, 7], [2, 4]]

# 搜索数字1
print(find_sublist_index(my_list, 1))  # 输出：1

# 搜索数字4
print(find_sublist_index(my_list, 4))  # 输出：2

# 搜索不存在的数字10
print(find_sublist_index(my_list, 10))  # 输出：None

代码说明 #

• 用enumerate()遍历嵌套列表，同时获取子列表的索引和内容
• 检查目标数字是否在当前子列表中，找到则立即返回对应索引
• 若遍历完所有子列表都未找到目标，返回None表示不存在

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