python中关于自己实现range方法

雪落千寒丶

1. class MyRange:
   
2. 
   
3.     def __init__(self, start, end=None, step=1):
   
4. 
   
5.         if end is not None:
   
6.             self.start = start
   
7.             self.end = end
   
8.         else:
   
9.             self.start = 0
   
10.             self.end = start
    
11. 
    
12.         self.step = step
    
13. 
    
14.     def __iter__(self):
    
15.         return self
    
16. 
    
17.     def __next__(self):
    
18. 
    
19.         while True:
    
20.             if self.step > 0:
    
21.                 if self.start < self.end:
    
22.                     current = self.start
    
23.                     self.start += self.step
    
24.                     return current
    
25. 
    
26.             elif self.step < 0:
    
27.                 if self.start > self.end:
    
28.                     current = self.start
    
29.                     self.start += self.step
    
30.                     return current
    
31. 
    
32.             else:
    
33.                 raise ValueError("MyRange() step argument must not be zero")
    
34. 
    
35. 
    
36. for i in MyRange(10):
    
37.     print(i)

复制代码

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

ba21

1.     def __next__(self):
   
2.             current = self.start
   
3.             if current > self.end:
   
4.                 raise StopIteration
   
5.             else:
   
6.                 self.start += self.step
   
7.                 return current
   

复制代码

1.     def __next__(self):
   
2. 
   
3.             if self.start > self.end:
   
4.                 raise StopIteration
   
5.             else:
   
6.                 self.start += self.step
   
7.                 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

雪落千寒丶

1.     def __next__(self):
   
2. 
   
3.         while True:
   
4.             if self.step > 0:
   
5.                 if self.start < self.end:
   
6.                     current = self.start
   
7.                     self.start += self.step
   
8.                     return current
   
9. 
   
10.                 elif self.start == self.end:
    
11.                     exit()
    
12. 
    
13.                 else:
    
14.                     raise ValueError("if step > 0, start must less than end.")
    
15. 
    
16.             elif self.step < 0:
    
17.                 if self.start > self.end:
    
18.                     current = self.start
    
19.                     self.start += self.step
    
20.                     return current
    
21. 
    
22.                 elif self.start == self.end:
    
23.                     exit()
    
24. 
    
25.                 else:
    
26.                     raise ValueError("if step < 0 , start must great than end.")
    
27. 
    
28.             else:
    
29.                 raise ValueError("MyRange() step argument must not be zero")

复制代码

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

ba21

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

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

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

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