One-liner untuk memeriksa apakah iterator menghasilkan setidaknya satu elemen?

Saat ini saya melakukan ini:

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

Tapi saya ingin ekspresi yang bisa saya tempatkan di dalam ifpernyataan sederhana . Apakah ada fitur bawaan yang dapat membuat kode ini terlihat tidak terlalu kikuk?

any()mengembalikan Falsejika sebuah iterable kosong, tetapi itu akan berpotensi mengulang semua item jika tidak. Saya hanya perlu untuk memeriksa item pertama.

Seseorang bertanya apa yang saya coba lakukan. Saya telah menulis sebuah fungsi yang mengeksekusi query SQL dan menghasilkan hasilnya. Terkadang ketika saya memanggil fungsi ini, saya hanya ingin tahu apakah kueri mengembalikan sesuatu dan membuat keputusan berdasarkan itu.

anytidak akan melampaui elemen pertama jika True. Jika iterator menghasilkan sesuatu yang salah, Anda dapat menulis any(True for _ in iterator).

Dalam Python 2.6+, jika nama sentinelterikat pada nilai yang tidak mungkin dihasilkan oleh iterator,

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

Jika Anda tidak tahu apa yang mungkin dihasilkan oleh iterator, buatlah sentinel Anda sendiri (misalnya di bagian atas modul Anda) dengan

sentinel = object()

Jika tidak, Anda dapat menggunakan, dalam peran sentinel, nilai apa pun yang "Anda ketahui" (berdasarkan pertimbangan aplikasi) yang tidak mungkin dihasilkan oleh iterator.

Ini tidak benar-benar lebih bersih, tetapi ini menunjukkan cara untuk mengemasnya dalam fungsi tanpa kehilangan:

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

Ini tidak benar-benar pythonic, dan untuk kasus tertentu, mungkin ada solusi yang lebih baik (tapi kurang umum), seperti default berikutnya .

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

Ini tidak umum karena Tidak Ada dapat menjadi elemen yang valid di banyak iterable.

kamu bisa memakai:

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

tapi itu agak tidak jelas untuk pembaca kode

Cara terbaik untuk melakukannya adalah dengan peekabledari more_itertools.

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

Berhati-hatilah jika Anda menyimpan referensi ke iterator lama, iterator itu akan menjadi maju. Anda harus menggunakan iterator baru yang dapat diintip sejak saat itu. Sungguh, meskipun, peekable berharap menjadi satu-satunya kode yang memodifikasi iterator lama itu, jadi Anda tidak boleh menyimpan referensi ke iterator lama yang ada di sekitar.

Bagaimana dengan:

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__ memperkirakan panjang list(it)- itu metode pribadi, meskipun:

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

Ini adalah pembungkus iterator berlebihan yang umumnya memungkinkan untuk memeriksa apakah ada item berikutnya (melalui konversi ke boolean). Tentu sangat tidak efisien.

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)

Keluaran:

False
[]
True
[1, 2, 3]

Sedikit terlambat, tapi ... Anda dapat mengubah iterator menjadi daftar dan kemudian bekerja dengan daftar itu:

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