El iterador de Python devuelve 'Ninguno' no deseado
¿Por qué mi iterador devuelve 'Ninguno' adicional en la salida? Para los parámetros / ejemplo a continuación, obtengo[None,4,None]
en lugar del deseado[4]
¿Alguien puede explicar por qué recibo el None adicional y cómo puedo solucionarlo? La impresión 'regresar' solo aparece una vez, así que supongo que solo se debe agregar un elemento a la función de devolución de llamadas.
código:
class Prizes(object):
def __init__(self,purchase,n,d):
self.purchase = purchase
self.length = len(purchase)
self.i = n-1
self.n = n
self.d = d
def __iter__(self):
return self
def __next__(self):
if self.i < self.length:
old = self.i
self.i += self.n
if (self.purchase[old])%(self.d) == 0:
print("returning")
return old+1
else:
raise StopIteration
def superPrize(purchases, n, d):
return list(Prizes(purchases, n, d))
purchases = [12, 43, 13, 465, 1, 13]
n = 2
d = 3
print(superPrize(purchases, n, d))
Salida:
returning
[None, 4, None]