python-3.x - 检查元素是否存在于列表列表中的第 N 个位置

Question

考虑一下这个列表:

ini_list = [[1, 2, 5, 10, 7], 
            [4, 3, 4, 3, 21], 
            [45, 65, 8, 8, 9]]

我需要能够查找列表列表的任何第三个元素中是否存在 4。因此，在这个示例中，我看到它确实存在于第二个列表 [4, 3, 4, 3, 21] 的位置 1 和 3 上，因此它将返回 true。

我可以做到这一点:

elem_to_find = 4
res1 = any(elem_to_find in sublist for sublist in ini_list)

但是通过这样做，我将搜索所有列表元素列表，而不仅仅是每个列表的第三个元素。什么是正确的方法？

Answer 1

你可以这样做:

res1 = any(elem_to_find == sublist[2] for sublist in ini_list if len(sublist) > 2)

这会将 elem_to_find 与 sublist[2] 进行比较，而不仅仅是 sublist，即第三个元素而不是整个列表。它还确保子列表的长度大于 2(否则第三个元素将不存在)。

下面是包含 2 个示例的完整代码:

ini_list = [[1, 2, 5, 10, 7], 
            [4, 3, 4, 3, 21], 
            [45, 65, 8, 8, 9]]

elem_to_find = 4
res1 = any(elem_to_find == sublist[2] for sublist in ini_list if len(sublist) > 2)

print(res1)
# Prints True

ini_list = [[1, 2, 5, 10, 7], 
            [4, 3, 3, 3, 21],  # No longer has 4 in 3rd position
            [45, 65, 8, 8, 9]]

res1 = any(elem_to_find == sublist[2] for sublist in ini_list if len(sublist) > 2)
print(res1)
# Prints False