用iter或yield来实现生成器的功能代码
时间: 2024-02-24 14:58:31 浏览: 80
Python generator生成器和yield表达式详解
下面是使用`iter`或`yield`实现一个简单的生成器的代码示例:
使用iter实现:
```python
class MyGenerator:
def __init__(self, start, end):
self.start = start
self.end = end
def __iter__(self):
self.current = self.start
return self
def __next__(self):
if self.current < self.end:
result = self.current
self.current += 1
return result
raise StopIteration
```
使用yield实现:
```python
def my_generator(start, end):
current = start
while current < end:
yield current
current += 1
```
这两个实现方式都可以用来生成一个从`start`到`end`的数字序列,您可以根据自己的需要进行修改和扩展。
阅读全文