自定义函数,实现一个简单地累加器
时间: 2024-01-27 11:03:49 浏览: 97
Spark代码笔记03——自定义排序、自定义分区、累加器、广播变量
好的,我可以帮你写一个简单的累加器函数。以下是代码示例:
```python
def accumulator():
total = 0
while True:
x = yield total
total += x
```
这个函数使用了 Python 的生成器(generator)来实现。它会持续地接受输入的数值并将其累加,每次累加后会返回当前的总和。你可以这样使用它:
```python
acc = accumulator()
next(acc) # 必须先调用一次 next(),让生成器准备好接受输入
print(acc.send(1)) # 输出 1,当前总和为 1
print(acc.send(2)) # 输出 3,当前总和为 1+2=3
print(acc.send(3)) # 输出 6,当前总和为 1+2+3=6
```
希望这个累加器函数能够满足你的需求!
阅读全文