Một lớp lót để kiểm tra xem một trình vòng lặp có tạo ra ít nhất một phần tử không?

Question

Hiện nay tôi đang làm điều này:

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

Nhưng tôi muốn một biểu thức mà tôi có thể đặt bên trong một tuyên bố đơn giản if. Có bất kỳ thứ gì được tích hợp sẵn để mã này trông ít vụng về không?

any() trả về False nếu có thể lặp lại trống, nhưng nó sẽ có khả năng lặp qua tất cả các mục nếu không. Tôi chỉ cần nó để kiểm tra mục đầu tiên.

---

Có người hỏi tôi đang cố gắng làm gì. Tôi đã viết một hàm thực hiện truy vấn SQL và mang lại kết quả của nó. Thỉnh thoảng khi tôi gọi hàm này, tôi chỉ muốn biết liệu truy vấn có trả về bất cứ điều gì và đưa ra quyết định dựa trên điều đó hay không.

Answer 1

any sẽ không vượt quá phần tử đầu tiên nếu đó là True. Trong trường hợp trình vòng lặp sinh ra một cái gì đó sai-ish bạn có thể viết any(True for _ in iterator).

Answer 2

__length_hint__ước tính chiều dài của list(it) - đó là phương pháp tư nhân, mặc dù:

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

Answer 3

Đây không phải là thực sự sạch hơn, nhưng nó cho thấy một cách để gói nó trong một hàm losslessly:

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 

Đây không phải là thực sự pythonic, và đối với trường hợp cụ thể, có lẽ tốt hơn (nhưng ít chung) các giải pháp, như mặc định next.

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

Đây không phải là chung vì Không có thể là yếu tố hợp lệ trong nhiều lần lặp lại.

Answer 4

bạn có thể sử dụng:

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

nhưng đó là một chút nonexplanatory cho đầu đọc mã

Answer 5

Đây là một wrapper iterator overkill mà thường cho phép để kiểm tra xem có một món đồ tiếp theo (thông qua chuyển đổi sang boolean) . Tất nhiên là không hiệu quả.

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) 

Output:

False 
[] 
True 
[1, 2, 3] 

Answer 6

Trong Python 2.6+, nếu tên sentinel được ràng buộc với một giá trị mà iterator không thể nào mang lại,

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

Nếu bạn không có ý tưởng về những gì mà trình vòng lặp có thể có khả năng sinh lợi, hãy tạo mã thông báo của riêng bạn (ví dụ: ở đầu mô-đun của bạn) với

sentinel = object() 

Nếu không, bạn có thể sử dụng, trong vai trò sentinel, bất kỳ giá trị nào bạn "biết" (dựa trên các cân nhắc của ứng dụng) mà trình vòng lặp không thể mang lại.

Answer 7

Một chút muộn, nhưng ... Bạn có thể biến các iterator vào một danh sách và sau đó làm việc với danh sách đó:

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

Answer 8

gì về:

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