Python：删除匹配某个条件的前 K 个元素

有时候我们需要从一个列表中删除前K个匹配特定条件的元素，Python中提供了不同的方法。

方法1： 使用循环

data = [2, 4, 1, 5, 3, 6, 8, 7]
k = 3
condition = 5

i = 0
while i < len(data) and k > 0:
    if data[i] == condition:
        del data[i]
        k -= 1
    else:
        i += 1

print(data) # Output: [2, 4, 1, 3, 6, 8, 7]

这种方法使用循环，遍历列表。当发现匹配的元素后，通过del操作符将其从列表中移除。

方法2：使用列表推导式

data = [2, 4, 1, 5, 3, 6, 8, 7]
k = 3
condition = 5

data = [i for i in data if i != condition][:len(data) - k]

print(data) # Output: [2, 4, 1, 3, 6, 8, 7]

这种方法使用了Python中的列表推导式。它根据条件筛选出符合要求的元素，并在其后使用切片操作符删除前k个元素。

建议使用列表推导式，因为它比循环更高效，并且代码更整洁。除此之外，使用切片操作符还可以确保列表中的元素顺序不会改变。