12

在大多数情况下，python 生成器是列表的良好替代品，我希望在其中检查空条件，这对于普通生成器是不可能的。我正在尝试编写一个包装器，该包装器将允许检查空条件，但仍然很懒惰并提供生成器的好处。

class mygen:
  def __init__(self,iterable):
    self.iterable = (x for x in iterable)
    self.peeked = False
    self.peek = None
  def __iter__(self):
    if self.peeked:
      yield self.peek
      self.peeked = False
    for val in self.iterable:
      if self.peeked:
        yield self.peek
        self.peeked = False
      yield val
    if self.peeked:
      yield self.peek
      self.peeked = False
  def __nonzero__(self):
    if self.peeked:
      return True
    try:
      self.peek = self.iterable.next()
      self.peeked = True
      return True
    except:
      return False
  1. 我认为它的行为就像一个普通的生成器一样正确。有没有我遗漏的角落案例？
  2. 这看起来并不优雅。有没有更好的pythonic方式来做同样的事情？

示例用法：

def get_odd(l):
    return mygen(x for x in l if x%2)

def print_odd(odd_nums):
  if odd_nums:
      print "odd numbers found",list(odd_nums)
  else:
      print "No odd numbers found"

print_odd(get_odd([2,4,6,8]))
print_odd(get_odd([2,4,6,8,7]))
4

2 回答 2

12

我通常不会实现这种生成器。有一种惯用的方法可以测试迭代器it 是否已用尽：

try:
    next_item = next(it)
except StopIteration:
    # exhausted, handle this case

用一些特定于项目的 LBYL 成语代替这个 EAFP 成语似乎令人困惑，而且根本没有好处。

也就是说，如果我真的想这样做，这就是我将如何实现它：

class MyIterator(object):
    def __init__(self, iterable):
        self._iterable = iter(iterable)
        self._exhausted = False
        self._cache_next_item()
    def _cache_next_item(self):
        try:
            self._next_item = next(self._iterable)
        except StopIteration:
            self._exhausted = True
    def __iter__(self):
        return self
    def next(self):
        if self._exhausted:
            raise StopIteration
        next_item = self._next_item
        self._cache_next_item()
        return next_item
    def __nonzero__(self):
        return not self._exhausted
于 2012-07-13T09:25:41.443 回答
4

用于itertools.tee实现非零测试，并在创建时简单地对其进行缓存：

from itertools import tee

class NonZeroIterable(object):
    def __init__(self, iterable):
        self.__iterable, test = tee(iter(iterable))
        try:
            test.next()
            self.__nonzero = True
        except StopIteration:
            self.__nonzero = False                 

    def __nonzero__(self):
        return self.__nonzero

    def __iter__(self):
        return self.__iterable

小演示：

>>> nz = NonZeroIterable('foobar')
>>> if nz: print list(nz)
... 
['f', 'o', 'o', 'b', 'a', 'r']
>>> nz2 = NonZeroIterable([])
>>> if not nz2: print 'empty'
... 
empty

这个版本的 NonZeroIterable 缓存了标志；因此，它只会告诉您迭代器在开始时是否为非空。如果您需要能够在其生命周期的其他点测试迭代，请改用Sven 的版本；那里的__nonzero__标志会在每次迭代后告诉你是否还有更多的项目要来。

关于你的例子的旁注

您的示例代码太简单了，不适合您的用例；您首先测试非空性（可能会遍历输入列表以搜索奇数），但无论如何都会耗尽整个迭代器。以下代码同样有效，并且不需要您发明打破 python 习语的方法：

def print_odd(odd_nums):
    odd_nums = list(odd_nums)
    if odd_nums:
        print "odd numbers found", odd_nums
    else:
        print "No odd numbers found"
于 2012-07-13T09:26:01.490 回答