python - 基于子列表过滤列表时如何避免硬编码？

Question

我有以下列表列表，我正在删除所有不包含 sl[0] 或 sl[1] 子列表的列表:

l = [
 ['s1', 's5', 's6', 's8', 's10', 's5', 's15', 's23'],
 ['s1', 's5', 's8', 's10', 's5', 's6', 's8', 's15', 's23'],
 ['s1', 's5', 's6', 's10', 's14', 's15', 's23']
]
sl = [['s5','s6','s8'],['s15', 's23']]

此代码针对 2 个子列表 sl[0] 和 sl[1] 进行了硬编码:

list(filter(lambda l : not(set(sl[0]+sl[1])-set(l)), l))

如何使这段代码更灵活？如果 sl 包含 5 个或 10 个或任何其他数量的子列表怎么办？

Answer 1

您可以使用 more-itertools 包来完成此操作。您需要在 l 的每个元素中检查 len(sl[i]) 的每个子序列。它会很慢，但可行。

我还创建了一个使用 partial 的工厂函数，这样您就可以弹出任何 sl 恰好是什么。这与您对 filter 的使用保持一致。

from more_itertools import windowed
from functools import partial

def ordered_check(x, sub):
    s = tuple(sub)
    for y in windowed(x, len(sub)):
        if s == y:
            return True
    return False

def ordered_check_all(x, subs):
    return all(ordered_check(x, sub) for sub in subs)

def check_factory(subs):
    return partial(ordered_check_all, subs=subs)

list(filter(check_factory(sl), l))
# returns
[['s1', 's5', 's6', 's8', 's10', 's5', 's15', 's23'],
 ['s1', 's5', 's8', 's10', 's5', 's6', 's8', 's15', 's23'],
 ['s1', 's5', 's6', 's10', 's14', 's15', 's23']]