写出用Python编写迭代器的代码。
以下是一个示例代码，它使用Python编写一个简单的迭代器，它将返回数字的平方:

```python
class Squares:
    def __init__(self, n):
        self.n = n
        self.current = 0
    
    def __iter__(self):
        return self
    
    def __next__(self):
        if self.current >= self.n:
            raise StopIteration
        else:
            square = self.current ** 2
            self.current += 1
            return square

# 使用迭代器
squares = Squares(5)
for square in squares:
    print(square)
```

当我们运行这个代码，输出结果应该是:

```
0
1
4
9
16
```