Einzeiler, um zu überprüfen, ob ein Iterator mindestens ein Element liefert?

Derzeit mache ich das:

try:
    something = iterator.next()
    # ...
except StopIteration:
    # ...

Aber ich möchte einen Ausdruck, den ich in eine einfache ifAussage einfügen kann. Ist etwas eingebaut, das diesen Code weniger ungeschickt aussehen lässt?

any()FalseGibt zurück, wenn eine Iterable leer ist. Wenn dies nicht der Fall ist, werden möglicherweise alle Elemente durchlaufen. Ich brauche es nur, um den ersten Artikel zu überprüfen.

Jemand fragt, was ich versuche zu tun. Ich habe eine Funktion geschrieben, die eine SQL-Abfrage ausführt und deren Ergebnisse liefert. Wenn ich diese Funktion aufrufe, möchte ich manchmal nur wissen, ob die Abfrage etwas zurückgegeben hat, und darauf basierend eine Entscheidung treffen.

anygeht nicht über das erste Element hinaus, wenn es wahr ist. Falls der Iterator etwas Falsches liefert, können Sie schreiben any(True for _ in iterator).

Wenn in Python 2.6+ der Name sentinelan einen Wert gebunden ist, den der Iterator möglicherweise nicht liefern kann,

if next(iterator, sentinel) is sentinel:
    print('iterator was empty')

Wenn Sie keine Ahnung haben, was der Iterator möglicherweise liefern könnte, erstellen Sie Ihren eigenen Sentinel (z. B. oben in Ihrem Modul) mit

sentinel = object()

Andernfalls könnten Sie in der Sentinel-Rolle jeden Wert verwenden, den Sie "kennen" (basierend auf Anwendungsüberlegungen), den der Iterator möglicherweise nicht liefern kann.

Dies ist nicht wirklich sauberer, aber es zeigt eine Möglichkeit, es verlustfrei in eine Funktion zu packen:

def has_elements(iter):
  from itertools import tee
  iter, any_check = tee(iter)
  try:
    any_check.next()
    return True, iter
  except StopIteration:
    return False, iter

has_el, iter = has_elements(iter)
if has_el:
  # not empty

Dies ist nicht wirklich pythonisch, und für bestimmte Fälle gibt es wahrscheinlich bessere (aber weniger allgemeine) Lösungen, wie die nächste Standardeinstellung.

first = next(iter, None)
if first:
  # Do something

Dies ist nicht allgemein, da None in vielen Iterables ein gültiges Element sein kann.

Sie können verwenden:

if zip([None], iterator):
    # ...
else:
    # ...

Für den Codeleser ist dies jedoch nicht erklärend

Der beste Weg, dies zu tun, ist mit einem peekablevon more_itertools.

from more_itertools import peekable
iterator = peekable(iterator)
if iterator:
    # Iterator is non-empty.
else:
    # Iterator is empty.

Passen Sie nur auf, wenn Sie den alten Iterator beibehalten, wird dieser Iterator erweitert. Von da an müssen Sie den neuen Peekable-Iterator verwenden. Wirklich erwartet Peekable jedoch, dass es das einzige Stück Code ist, das diesen alten Iterator modifiziert, so dass Sie die Refs des alten Iterators, der herumliegt, sowieso nicht behalten sollten.

Wie wäre es mit:

In [1]: i=iter([])

In [2]: bool(next(i,False))
Out[2]: False

In [3]: i=iter([1])

In [4]: bool(next(i,False))
Out[4]: True

__length_hint__ schätzt die Länge von list(it)- es ist jedoch eine private Methode:

x = iter( (1, 2, 3) )
help(x.__length_hint__)
      1 Help on built-in function __length_hint__:
      2 
      3 __length_hint__(...)
      4     Private method returning an estimate of len(list(it)).

Dies ist ein Overkill-Iterator-Wrapper, mit dem im Allgemeinen überprüft werden kann, ob ein nächstes Element vorhanden ist (durch Konvertierung in einen Booleschen Wert). Natürlich ziemlich ineffizient.

class LookaheadIterator ():

    def __init__(self, iterator):
        self.__iterator = iterator
        try:
            self.__next      = next (iterator)
            self.__have_next = True
        except StopIteration:
            self.__have_next = False

    def __iter__(self):
        return self

    def next (self):
        if self.__have_next:
            result = self.__next
            try:
                self.__next      = next (self.__iterator)
                self.__have_next = True
            except StopIteration:
                self.__have_next = False

            return result

        else:
            raise StopIteration

    def __nonzero__(self):
        return self.__have_next

x = LookaheadIterator (iter ([]))
print bool (x)
print list (x)

x = LookaheadIterator (iter ([1, 2, 3]))
print bool (x)
print list (x)

Ausgabe:

False
[]
True
[1, 2, 3]

Ein bisschen spät, aber ... Sie könnten den Iterator in eine Liste verwandeln und dann mit dieser Liste arbeiten:

# Create a list of objects but runs out the iterator.
l = [_ for _ in iterator]

# If the list is not empty then the iterator had elements; else it was empty.
if l :
    pass # Use the elements of the list (i.e. from the iterator)
else :
    pass # Iterator was empty, thus list is empty.