Одне вкладиш, щоб перевірити, чи дає ітератор щонайменше один елемент?

На даний момент я роблю це:

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

але це трохи незрозуміло для читача коду

Найкращий спосіб зробити це за допомогою a 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.