Como verificar se todos os elementos de uma lista correspondem a uma condição?

Eu tenho uma lista que consiste em como 20000 listas. Eu uso o terceiro elemento de cada lista como uma bandeira. Quero fazer algumas operações nesta lista, desde que pelo menos o sinalizador de um elemento seja 0, é como:

my_list = [["a", "b", 0], ["c", "d", 0], ["e", "f", 0], .....]

No começo, todos os sinalizadores são 0. Uso um loop while para verificar se pelo menos o sinalizador de um elemento é 0:

def check(list_):
    for item in list_:
        if item[2] == 0:
            return True
    return False

Se check(my_list)retornar True, continuarei trabalhando na minha lista:

while check(my_list):
    for item in my_list:
        if condition:
            item[2] = 1
        else:
            do_sth()

Na verdade, eu queria remover um elemento em my_list enquanto iterava sobre ele, mas não tenho permissão para remover itens enquanto iteramos sobre ele.

My_list original não tinha sinalizadores:

my_list = [["a", "b"], ["c", "d"], ["e", "f"], .....]

Como não consegui remover elementos enquanto iterava sobre ele, inventei esses sinalizadores. Mas o my_listcontém muitos itens, e o whileloop lê todos eles em cada forloop e consome muito tempo! Você tem alguma sugestão?

A melhor resposta aqui é usar all(), que é a base para esta situação. Combinamos isso com uma expressão de gerador para produzir o resultado desejado de forma limpa e eficiente. Por exemplo:

>>> items = [[1, 2, 0], [1, 2, 0], [1, 2, 0]]
>>> all(flag == 0 for (_, _, flag) in items)
True
>>> items = [[1, 2, 0], [1, 2, 1], [1, 2, 0]]
>>> all(flag == 0 for (_, _, flag) in items)
False

Observe que all(flag == 0 for (_, _, flag) in items)é diretamente equivalente a all(item[2] == 0 for item in items), é apenas um pouco mais agradável de ler neste caso.

E, para o exemplo de filtro, uma compreensão da lista (é claro, você pode usar uma expressão geradora, quando apropriado):

>>> [x for x in items if x[2] == 0]
[[1, 2, 0], [1, 2, 0]]

Se você deseja verificar pelo menos um elemento é 0, a melhor opção é usar o any()que for mais legível:

>>> any(flag == 0 for (_, _, flag) in items)
True

Se você deseja verificar se algum item da lista viola uma condição, use all:

if all([x[2] == 0 for x in lista]):
    # Will run if all elements in the list has x[2] = 0 (use not to invert if necessary)

Para remover todos os elementos que não correspondem, use filter

# Will remove all elements where x[2] is 0
listb = filter(lambda x: x[2] != 0, listb)

Você pode usar as ferramentas do itertools desse jeito, ele será interrompido assim que uma condição for atendida e falhar na sua declaração. O método oposto seria suspenso

for x in itertools.takewhile(lambda x: x[2] == 0, list)
    print x

Outra maneira de usar itertools.ifilter. Isso verifica a veracidade e o processo (usandolambda )

Amostra-

for x in itertools.ifilter(lambda x: x[2] == 0, my_list):
    print x

dessa maneira é um pouco mais flexível do que usar all():

my_list = [[1, 2, 0], [1, 2, 0], [1, 2, 0]]
all_zeros = False if False in [x[2] == 0 for x in my_list] else True
any_zeros = True if True in [x[2] == 0 for x in my_list] else False

ou mais sucintamente:

all_zeros = not False in [x[2] == 0 for x in my_list]
any_zeros = 0 in [x[2] for x in my_list]