python:如果在另一个列表中匹配,则将嵌套列表子集化

你可以 zip m s 拉链

list(zip(*(c for c in zip(*m) if c[1] in s)))

这将返回:

[(1, 3), ('a', 'c')]

拉链 到 list

list(map(list, list(zip(*(c for c in zip(*m) if c[1] in s)))))

这将返回:

[[1, 3], ['a', 'c']]

你可以选择 如果中有许多项,则首先设置以提高查找效率 s :

s = set(s)

m2 = [[], []]

for i in range(len(m[1])):
    if m[1][i] in s:
        m2[0].append(m[0][i])
        m2[1].append(m[1][i])