删除与Python列表中的条件匹配的前N个项

Question

如果我有一个函数matchCondition(x),如何删除n匹配该条件的Python列表中的第一项？

一种解决方案是迭代每个项目,将其标记为删除(例如,通过将其设置为None),然后用理解来过滤列表.这需要对列表进行两次迭代并改变数据.是否有更惯用或更有效的方法来做到这一点？

n = 3

def condition(x):
    return x < 5

data = [1, 10, 2, 9, 3, 8, 4, 7]
out = do_remove(data, n, condition)
print(out)  # [10, 9, 8, 4, 7] (1, 2, and 3 are removed, 4 remains)

Answer 1

使用itertools.filterfalse和的一种方式itertools.count:

from itertools import count, filterfalse

data = [1, 10, 2, 9, 3, 8, 4, 7]
output = filterfalse(lambda L, c=count(): L < 5 and next(c) < 3, data)

然后list(output),给你:

[10, 9, 8, 4, 7]

Answer 2

编写一个生成器,它接受可迭代,条件和数量下降.迭代数据并生成不符合条件的项目.如果满足条件,则递增计数器并且不产生该值.一旦计数器达到您想要下降的数量,总是产生物品.

def iter_drop_n(data, condition, drop):
    dropped = 0

    for item in data:
        if dropped >= drop:
            yield item
            continue

        if condition(item):
            dropped += 1
            continue

        yield item

data = [1, 10, 2, 9, 3, 8, 4, 7]
out = list(iter_drop_n(data, lambda x: x < 5, 3))

这不需要额外的列表副本,只迭代列表一次,并且只为每个项目调用一次条件.除非您确实想要查看整个列表,否则不要list对结果进行调用并直接迭代返回的生成器.

Answer 3

接受的答案对我来说有点太神奇了.这里的流程有望更清晰:

def matchCondition(x):
    return x < 5


def my_gen(L, drop_condition, max_drops=3):
    count = 0
    iterator = iter(L)
    for element in iterator:
        if drop_condition(element):
            count += 1
            if count >= max_drops:
                break
        else:
            yield element
    yield from iterator


example = [1, 10, 2, 9, 3, 8, 4, 7]

print(list(my_gen(example, drop_condition=matchCondition)))

它类似于大卫答案中的逻辑,但是不是检查每一步的丢弃计数,我们只是将循环的其余部分短路.

注意:如果您没有yield from可用的,只需将其替换为另一个用于循环其余项目iterator.