python - 如何测试整数是否在列表的子列表中

Question

我想测试 3 是否是元素的第一个数字（整数或子列表的第一个），如下所示：

lst=[2, [3,6], [4,1,7]]
3 in lst

结果应该是 True 因为 3 是 [3,6] 的第一个元素。

顺便说一句：我的数据集不会使我的列表像 [3, [3,7]] （单独和在子列表中）

标签： python

Answer 1

假设没有子子列表：

l=[2, [3,6], [4,1,7]]
first_elements = [i[0] if isinstance(i, list) else i for i in l]  # [2, 3, 4]
print(3 in first_elements)

输出：

True