ซับเดียวเพื่อตรวจสอบว่าตัววนซ้ำให้องค์ประกอบอย่างน้อยหนึ่งองค์ประกอบหรือไม่?

ตอนนี้ฉันกำลังทำสิ่งนี้:

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

นี่ไม่ใช่ pythonic จริงๆและในบางกรณีอาจมีโซลูชันที่ดีกว่า (แต่ทั่วไปน้อยกว่า) เช่นค่าเริ่มต้นถัดไป

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

นี่ไม่ใช่เรื่องทั่วไปเนื่องจากไม่มีองค์ประกอบที่ถูกต้องในการทำซ้ำหลายรายการ

คุณสามารถใช้ได้:

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

แต่มันไม่มีอะไรจำเป็นสำหรับโปรแกรมอ่านโค้ด

วิธีที่ดีที่สุดที่จะทำคือมีจากpeekablemore_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)).

นี่คือกระดาษห่อตัววนซ้ำแบบ overkill ซึ่งโดยทั่วไปจะช่วยให้ตรวจสอบว่ามีรายการถัดไปหรือไม่ (ผ่านการแปลงเป็นบูลีน) ไม่มีประสิทธิภาพแน่นอน

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]

สายไปหน่อย แต่ ... คุณสามารถเปลี่ยน iterator เป็นรายการจากนั้นทำงานกับรายการนั้น:

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