如何測試一個整數是否在列表的子列表中

Question

我想測試 3 是否是元素的第一個數字（整數或子列表的第一個），如下所示：

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

結果應該是 True，因為 3 是 [3,6] 的第一個元素。

順便說一句：我的數據集不會讓我的列表像 [3, [3,7]] （單獨和在子列表中）

Answer 1

你可以用一個非常簡單的遞歸函數來做到這一點，比如：

l =[2, [3,6], [4,1,7]]

def inList(l, n):
    if isinstance(l, list):
        return l[0] == n or any(inList(member, n) for member in l[1:])
    else:
        return False

inList(l, 3) # True
inList(l, 9) # False
inList(l, 2) # True

這也具有深入嵌套列表的優點：

l =[2, [3,6], [4,1,[9,[5]], 7]]
inList(l, 5) # True

Answer 2

假設沒有子子列表：

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

Answer 3

您可以遍歷列表，並檢查子元素sub是列表還是整數，並返回您想要的結果：

L = [2, [3,6], [4,1,7]]
num = 2
res = False
for sub in L:
    if isinstance(sub, list):
        if num in sub:
            res = True
            break
    else:
        if num == sub:
            res = True

print(res)