tutorial - Elimine los primeros N elementos que coincidan con una condición en una lista de Python

tutorial django

Question

Si tengo una función matchCondition(x) , ¿cómo puedo eliminar los primeros n elementos de una lista de Python que coinciden con esa condición?

Una solución es iterar sobre cada elemento, marcarlo para su eliminación (por ejemplo, configurándolo en None ) y luego filtrar la lista con una comprensión. Esto requiere iterar sobre la lista dos veces y muta los datos. ¿Hay alguna forma más idiomática o eficiente de hacer esto?

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

Comenzando Python 3.8 , y la introducción de expresiones de asignación (PEP 572) ( := operador), podemos usar e incrementar una variable dentro de una comprensión de lista:

# items = [1, 10, 2, 9, 3, 8, 4, 7]
total = 0
[x for x in items if not (x < 5 and (total := total + 1) <= 3)]
# [10, 9, 8, 4, 7]

Esta:

• Inicializa una variable total a 0 que simbolizará el número de ocurrencias previamente coincidentes dentro de la comprensión de la lista
• Comprueba cada elemento si ambos:
  • coincide con la condición de exclusión ( x < 5 )
  • y si aún no hemos descartado más de la cantidad de elementos que queremos filtrar por:
    • Incremento total ( total := total + 1 ) a través de una expresión de asignación
    • y al mismo tiempo comparando el nuevo valor del total con el número máximo de elementos a descartar ( 3 )

Answer 2

Escriba un generador que tome el iterable, una condición y una cantidad para soltar. Itere sobre los datos y produzca elementos que no cumplan con la condición. Si se cumple la condición, incremente un contador y no produzca el valor. Siempre rinda artículos una vez que el contador alcance la cantidad que desea soltar.

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))

Esto no requiere una copia adicional de la lista, solo itera sobre la lista una vez y solo llama a la condición una vez para cada elemento. A menos que realmente quiera ver la lista completa, deje la llamada de la list en el resultado e itere directamente sobre el generador devuelto.

Answer 3

La respuesta aceptada fue demasiado mágica para mi gusto. Aquí hay uno donde el flujo es un poco más claro de seguir:

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)))

Es similar a la lógica en la respuesta de davidism , pero en lugar de verificar que el conteo de caídas se excede en cada paso, simplemente cortocircuitamos el resto del ciclo.

Nota: Si no tiene yield from disponible, simplemente reemplácelo con otro bucle for sobre los elementos restantes en el iterator .

Answer 4

Python directo:

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

def matchCondition(x):
    return x < 5

c = 1
l = []
for x in data:
    if c > N or not matchCondition(x):
        l.append(x)
    else:
        c += 1

print(l)

Esto puede convertirse fácilmente en un generador si lo desea:

def filter_first(n, func, iterable):
    c = 1
    for x in iterable:
        if c > n or not func(x):
            yield x
        else:
            c += 1

print(list(filter_first(N, matchCondition, data)))

Answer 5

Si se requiere mutación:

def do_remove(ls, N, predicate):
    i, delete_count, l = 0, 0, len(ls)
    while i < l and delete_count < N:
        if predicate(ls[i]):
           ls.pop(i) # remove item at i
           delete_count, l = delete_count + 1, l - 1 
        else:
           i += 1
    return ls # for convenience

assert(do_remove(l, N, matchCondition) == [10, 9, 8, 4, 7])

Answer 6

Una forma de usar itertools.filterfalse e 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)

Luego, la list(output) le da:

[10, 9, 8, 4, 7]

Answer 7

Usando listas de comprensión:

n = 3
data = [1, 10, 2, 9, 3, 8, 4, 7]
count = 0
def counter(x):
    global count
    count += 1
    return x

def condition(x):
    return x < 5

filtered = [counter(x) for x in data if count < n and condition(x)]

Esto también dejará de verificar la condición después de que se encuentren n elementos gracias al cortocircuito booleano.