Однострочный, чтобы проверить, дает ли итератор хотя бы один элемент?

Сейчас делаю вот что:

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

Но мне нужно выражение, которое я могу поместить в простое ifутверждение. Есть ли что-нибудь встроенное, что сделало бы этот код менее неуклюжим?

any()возвращается, Falseесли итерация пуста, но потенциально может перебирать все элементы, если это не так. Мне это нужно только для проверки первого пункта.

Кто-то спрашивает, что я пытаюсь сделать. Я написал функцию, которая выполняет SQL-запрос и возвращает его результаты. Иногда, когда я вызываю эту функцию, я просто хочу узнать, вернул ли запрос что-нибудь, и принять решение на основе этого.

anyне выйдет за пределы первого элемента, если он True. Если итератор выдает что-то ложное, вы можете написать any(True for _ in iterator).

В Python 2.6+, если имя sentinelпривязано к значению, которое итератор не может выдать,

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

Если вы не знаете, что может дать итератор, сделайте свой собственный дозорный (например, в верхней части модуля) с помощью

sentinel = object()

В противном случае вы можете использовать в роли дозорного любое значение, которое вам «известно» (исходя из соображений приложения), которое итератор не может выдать.

Это не совсем чище, но показывает способ упаковать его в функцию без потерь:

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

На самом деле это не питонизм, и для конкретных случаев, вероятно, есть лучшие (но менее общие) решения, такие как следующее по умолчанию.

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

Это не является общим, потому что None может быть допустимым элементом во многих итерациях.

ты можешь использовать:

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

но это немного необъяснимо для читателя кода

Лучший способ сделать это - использовать peekablefrom more_itertools.

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

Просто будьте осторожны, если вы сохранили ссылки на старый итератор, этот итератор будет расширяться. С этого момента вы должны использовать новый просматриваемый итератор. На самом деле, peekable ожидает быть единственным фрагментом кода, изменяющим этот старый итератор, так что вам все равно не следует хранить ссылки на старый итератор.

Что о:

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__ оценивает длину list(it)- это частный метод, однако:

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

Это избыточная оболочка итератора, которая обычно позволяет проверять, есть ли следующий элемент (путем преобразования в логическое значение). Конечно довольно неэффективно.

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)

Вывод:

False
[]
True
[1, 2, 3]

Немного поздно, но ... Вы можете превратить итератор в список, а затем работать с этим списком:

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