Wie überprüfe ich, ob alle Elemente einer Liste einer Bedingung entsprechen?

Ich habe eine Liste, die aus etwa 20000 Listen besteht. Ich verwende das dritte Element jeder Liste als Flag. Ich möchte einige Operationen auf dieser Liste ausführen, solange das Flag mindestens eines Elements 0 ist, es ist wie:

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

Am Anfang sind alle Flags 0. Ich verwende eine While-Schleife, um zu prüfen, ob das Flag mindestens eines Elements 0 ist:

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

Wenn check(my_list) kehrt zurück Truedann arbeite ich weiter an meiner Liste:

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

Eigentlich wollte ich ein Element in my_list entfernen, während ich darüber iteriere, aber ich darf keine Elemente entfernen, während ich darüber iteriere.

Ursprüngliche my_list hatte keine Flags:

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

Da ich beim Iterieren keine Elemente entfernen konnte, habe ich diese Flags erfunden. Aber die my_list enthält viele Elemente, und while Schleife liest alle von ihnen bei jedem for Schleife, und es kostet viel Zeit! Haben Sie Vorschläge?

Die beste Antwort hier ist zu verwenden all(), die für diese Situation integriert ist. Wir kombinieren dies mit a Generatorausdruck um sauber und effizient das gewünschte Ergebnis zu erzielen. Zum Beispiel:

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

Beachten Sie, dass all(flag == 0 for (_, _, flag) in items) ist direkt äquivalent zu all(item[2] == 0 for item in items)es ist in diesem Fall nur etwas schöner zu lesen.

Und für das Filterbeispiel ein Listenverständnis (natürlich könnten Sie gegebenenfalls einen Generatorausdruck verwenden):

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

Wenn Sie überprüfen möchten, ob mindestens ein Element 0 ist, ist die bessere Option zu verwenden any() was besser lesbar ist:

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

Wenn Sie überprüfen möchten, ob ein Element in der Liste gegen eine Bedingung verstößt, verwenden Sie 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)

Verwenden Sie zum Entfernen aller nicht übereinstimmenden Elemente filter

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

Sie könnten das takewhile von itertools so verwenden, dass es aufhört, sobald eine Bedingung erfüllt ist, die Ihre Aussage nicht erfüllt. Die entgegengesetzte Methode wäre dropwhile

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

Dieser Weg ist etwas flexibler als die Verwendung 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

oder kürzer:

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

Eine andere Art der Verwendung itertools.ifilter. Dies überprüft die Richtigkeit und den Prozess (unter Verwendung von lambda)

Probe-

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