генератор на Python с проверкой на пустое состояние
генераторы 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
- Я думаю, что он ведет себя правильно, как простой генератор. Есть ли случай Я пропал?
- Это не выглядит элегантно. Есть лучше более подходящие для Python способ сделать то же самое?
пример использования:
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]))
2 ответов
Я бы обычно не реализовывал этот вид
генератора. Существует идиоматический способ проверить, если итератор it
изнемогает:
try:
next_item = next(it)
except StopIteration:
# exhausted, handle this case
замена этой идиомы EAFP некоторой специфичной для проекта идиомой LBYL кажется сбивает с толку и совсем не приносит пользы.
тем не менее, вот как я бы реализовал это, если бы я действительно хотел:
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
использовать 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 кэширует флаг; он таким образом только сообщает вам, был ли итератор непустым в начале. Если вам нужно протестировать iterable в других точках жизненного цикла, используйте Свена!-Вместо -7-->; здесь __nonzero__
флаг вам скажу после каждого итерация если есть больше пунктов, чтобы прийти еще.
примечание на вашем примере
ваш пример кода слишком прост и не является хорошим аргументом для вашего usecase; вы сначала проверяете непустоту (которая потенциально повторяется над входным списком для поиска нечетного числа), но затем исчерпываете весь итератор в любом случае. Следующий код будет столь же эффективным и не потребует от вас изобретать способы разбить идиомы 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"