Iterador Python retornando indesejados 'Nenhum'
Por que meu iterador está retornando extra 'Nenhum' na saída. Para os parâmetros / exemplo abaixo, estou recebendo[None,4,None]
em vez do desejado[4]
Alguém pode explicar por que estou recebendo o Extra extra e como posso corrigi-lo? A impressão 'retornando' aparece apenas uma vez; portanto, estou assumindo que apenas um item deve ser anexado à função de retorno da chamada.
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))
Resultado:
returning
[None, 4, None]