檢查發電機真的在一行中產生的東西

Question

許多圖書館的功能我用回generator！而非list■當結果是一些收藏。

有時候，我只是想檢查如果結果是空的沒有了，當然我不會寫這樣的：

result = return_generator() 
    if result: 
     print 'Yes, the generator did generate something!' 

現在我來了一行代碼是沒有解決這個問題消費發生器：

result = return_generator() 
    if zip("_", result): 
     print 'Yes, the generator did generate something!' 

我不知道是否有簡潔的方式來解決這個問題在一行？

Answer 1

這zip東西吃的第一個項目產生，所以它不是一個好主意爲好。

如果發生器由獲得並保持它，直到得到所需的項目只能檢測。以下課程將幫助你這樣做。

如果需要，它會從迭代器中獲取一個項目並保留它。

如果要求emptyness（if myiterwatch: ...），它會嘗試獲取並返回如果能得到一個。

如果問下一個項目，它將返回檢索到的一個或一個新的。

class IterWatch(object): 
    def __init__(self, it): 
     self.iter = iter(it) 
     self._pending = [] 
    @property 
    def pending(self): 
     try: 
      if not self._pending: 
       # will raise StopIteration if exhausted 
       self._pending.append(next(self.iter)) 
     except StopIteration: 
      pass # swallow this 
     return self._pending 
    def next(self): 
     try: 
      return self.pending.pop(0) 
     except IndexError: 
      raise StopIteration 
    __next__ = next # for Py3 
    def __iter__(self): return self 
    def __nonzero__(self): 
     # returns True if we have data. 
     return not not self.pending 
     # or maybe bool(self.pending) 
    __bool__ = __nonzero__ # for Py3 

---

這解決了一個非常通用的方式的問題。如果你有，你只是想測試一個迭代器，你可以使用

guard = object() 
result = return_generator() 
if next(result, guard) is not guard: 
    print 'Yes, the generator did generate something!' 

next(a, b)回報b如果迭代器a耗盡。所以如果它在我們的情況下返回警衛，那麼它不會產生任何東西，否則它會發生。

但你zip()的做法是完全合法的，以及...

Answer 2

而不是返回發電機，只是使用它？以下是可能返回結果或不返回結果的生成器示例 - 如果有結果採取行動，如果沒有結果，則不採取任何行動。

#!/usr/bin/python 

import time 

def do_work(): 
    if int(time.time()) % 2: 
     for a in xrange(0,100): 
      yield a 

for thing in do_work(): 
    print thing 

Answer 3

這裏使用itertools單向三通功能，複製發電機：

from itertools import tee 
a, b = tee(xrange(0)) 
try: 
    next(a) 
    print list(b) 
except StopIteration: 
    print "f1 was empty" 


a, b = tee(xrange(3)) 
try: 
    next(a) 
    print list(b) 
except StopIteration: 
    print "f2 was empty" 

>>> 
[0, 1, 2, 3] 
f2 was empty 

Answer 4

這裏有適合自己情況的兩個最簡單的解決方案我可以想到：

def f1(): 
    return (i for i in range(10)) 

def f2(): 
    return (i for i in range(0)) 


def has_next(g): 
    try: 
     from itertools import chain 
     return chain([g.next()],g) 
    except StopIteration: 
     return False 

g = has_next(f1()) 
if g: 
    print list(g) 

g = has_next(f2()) 
if g: 
    print list(g) 


def has_next(g): 
    for i in g: 
     return chain([i],g) 
    return False 

g = has_next(f1()) 
if g: 
    print list(g) 

g = has_next(f2()) 
if g: 
    print list(g)