Python实现: 利用迭代器的原理构造一个具有与range()相同功能的迭代器。命名为myrange类 利用迭代器的原理构造一个具有与zip()相同功能的迭代器,命名为myzip类
时间: 2024-02-10 14:10:47 浏览: 27
好的,我会用 Python 实现。
首先是 `myrange` 类:
```python
class myrange:
def __init__(self, start, stop=None, step=1):
if stop is None:
stop = start
start = 0
self.start = start
self.stop = stop
self.step = step
self.current = start
def __iter__(self):
return self
def __next__(self):
if self.step > 0 and self.current >= self.stop:
raise StopIteration
elif self.step < 0 and self.current <= self.stop:
raise StopIteration
else:
result = self.current
self.current += self.step
return result
```
`myrange` 类接受与 `range()` 函数相同的参数,即起始值、终止值和步长。它的 `__iter__` 方法返回自身,因此它是可迭代的。`__next__` 方法计算下一个值并返回它,直到达到终止值或步长为零。
接下来是 `myzip` 类:
```python
class myzip:
def __init__(self, *args):
self.iterators = [iter(arg) for arg in args]
def __iter__(self):
return self
def __next__(self):
try:
return tuple(next(iterator) for iterator in self.iterators)
except StopIteration:
raise StopIteration
```
`myzip` 类接受任意数量的可迭代对象,并为每个对象创建一个迭代器。它的 `__iter__` 方法返回自身,因此它是可迭代的。`__next__` 方法从每个迭代器中取出一个元素,并返回一个元组,直到任何一个迭代器抛出 `StopIteration` 异常。此时,它也抛出 `StopIteration` 异常以终止迭代。
使用这两个类的例子:
```python
# 使用 myrange
for i in myrange(5):
print(i) # 0 1 2 3 4
# 使用 myzip
a = [1, 2, 3]
b = ['a', 'b', 'c']
for x, y in myzip(a, b):
print(x, y) # 1 a 2 b 3 c
```