如何判断元素是否为嵌套列表中任意子列表指定索引位置的元素

你需要的功能可以通过any()函数搭配生成器表达式实现，该方案支持短路求值，找到第一个匹配项就会立即终止遍历，不需要遍历完整个嵌套列表。

基础实现代码 #

l_list = [['a', 2, True], ['b', 1, False], ['c', 5, True]]
target_val = 'a'
check_pos = 0 # 要检查的子列表索引位置，这里对应第一个元素

is_exist = any(sublist[check_pos] == target_val for sublist in l_list)
if is_exist:
    print("Yes")

兼容子列表长度不足的优化实现 #

如果存在部分子列表长度小于你要检查的索引位置，可以加长度判断避免抛出索引越界异常：

is_exist = any(
    len(sublist) > check_pos and sublist[check_pos] == target_val 
    for sublist in l_list
)

高频查询场景优化 #

如果你需要多次执行这类查询，可以提前将所有子列表指定索引的元素预存为集合，后续所有查询都是O(1)复杂度，不需要再遍历原列表：

# 仅首次需要遍历一次原列表构建索引集合
pos_set = {sublist[check_pos] for sublist in l_list if len(sublist) > check_pos}

# 后续任意次数查询都不需要遍历
print('a' in pos_set) # 输出True
print('d' in pos_set) # 输出False

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