Проблема выхода из генератора во время итерации цикла for - PullRequest
Новая
       8

Проблема выхода из генератора во время итерации цикла for

0 голосов
/ 23 октября 2019

Назначен для создания класса связанного списка с нуля с использованием генератора для итерации по списку, но у меня возникла проблема при отладке этой проблемы: 

class LinkedList:

    def __init__(self, data = None):
        self.node = Node(data)
        self.first = self.node 
        self.last = self.node
        self.n = 1

    def append(self, data = None):
        new = Node(data)
        self.last.next = new
        self.last = new
        self.n += 1

    def __iter__(self):
        self.index = self.first
        return self.generator()

    def generator(self):
        for i in range(0, self.n):
            yield self.index
            self.index = self.index.next

Один из тестов, которые проходит текущий генераторis: 

for n in a:
    print n

# which should and does output

0
1
2

Однако в этом случае он не может вывести правильные значения 

for n in a:
    if n == 2:
        break
    else:
        print n

# Outputs 0 1 2, instead of:

0
1

Я думаю, что мое понимание генераторов отсутствует, и я хотел бы получить любую помощь, которую вам, ребята, придетсяпредложение!

1 Ответ

0 голосов
/ 23 октября 2019
  • Вы не должны использовать n == 2 в качестве условия прерывания.

Попробуйте: 

# Assume your defined class `Node.py` like this:
class Node:
    def __init__(self, data=None):
        self.val = data
        self.next = None

# I didn't modify your `LinkedList.py`
class LinkedList:

    def __init__(self, data=None):
        self.node = Node(data)
        self.first = self.node
        self.last = self.node
        self.n = 1

    def append(self, data=None):
        new = Node(data)
        self.last.next = new
        self.last = new
        self.n += 1

    def __iter__(self):
        self.index = self.first
        return self.generator()

    def generator(self):
        for i in range(0, self.n):
            yield self.index
            self.index = self.index.next


if __name__ == '__main__':
    ass = LinkedList(0)
    ass.append(1)
    ass.append(2)
    ass.append(3)

    for n in ass:
        if n.val == 2:  # Notice here
            break
        else:
            print(n.val)  # And here
...