python中关于自己实现range方法

雪落千寒丶

class MyRange:

    def __init__(self, start, end=None, step=1):

        if end is not None:
            self.start = start
            self.end = end
        else:
            self.start = 0
            self.end = start

        self.step = step

    def __iter__(self):
        return self

    def __next__(self):

        while True:
            if self.step > 0:
                if self.start < self.end:
                    current = self.start
                    self.start += self.step
                    return current

            elif self.step < 0:
                if self.start > self.end:
                    current = self.start
                    self.start += self.step
                    return current

            else:
                raise ValueError("MyRange() step argument must not be zero")


for i in MyRange(10):
    print(i)

能实现range的效果 可是执行程序后打印完后 程序一直处于执行状态 无法结束 该如何优化代码呢？

ba21

    def __next__(self):
            current = self.start
            if current > self.end:
                raise StopIteration
            else:
                self.start += self.step
                return current

    def __next__(self):

            if self.start > self.end:
                raise StopIteration
            else:
                self.start += self.step
                return self.start 

雪落千寒丶

ba21 发表于 2018-7-2 16:58

你这个代码有问题呀

ba21

雪落千寒丶 发表于 2018-7-2 18:59
你这个代码有问题呀

有问题你不会自己改？什么都要人家帮你弄好，你自己还学什么。
真心不想多说

雪落千寒丶

ba21 发表于 2018-7-2 19:44
有问题你不会自己改？什么都要人家帮你弄好，你自己还学什么。
真心不想多说

我只是给你指正错误，你不接受就算了

雪落千寒丶

iwanna 发表于 2018-7-2 22:00
因为你定义了一个死循环啊，range是一个生成器，定义__next__方法就可以不断的生成下一个值，到头就raise S ...

恩 我知道是死循环的问题 可是不加while true 他会一直返回None

雪落千寒丶

    def __next__(self):

        while True:
            if self.step > 0:
                if self.start < self.end:
                    current = self.start
                    self.start += self.step
                    return current

                elif self.start == self.end:
                    exit()

                else:
                    raise ValueError("if step > 0, start must less than end.")

            elif self.step < 0:
                if self.start > self.end:
                    current = self.start
                    self.start += self.step
                    return current

                elif self.start == self.end:
                    exit()

                else:
                    raise ValueError("if step < 0 , start must great than end.")

            else:
                raise ValueError("MyRange() step argument must not be zero")

OK OK 问题解决了 现在再考虑考虑优化代码

ba21

雪落千寒丶 发表于 2018-7-3 16:35
我只是给你指正错误，你不接受就算了

用得着你指正吗？
我的代码有问题还是你有代码有问题？__next__方法内还用while 麻烦你还是把基础打牢。别乱指错误。

好心给你指点不接受还乱指错误；小小问题都不会纠正，还非得要别人给你上个全正确代码

敢问我的代码何处有问题？？？？