python - 将产量分成小于 Python 阈值的部分

Question

假设我有一个数字 n=22500 和一个阈值 t=10000，我想要一个函数产生:10000， 10000，2500。

我该怎么做？我的尝试无效，因为一旦超过 20000，它就会返回大于 10000 的数字。

def chunk(number):
    steps = 10000
    if number > steps:
        yield steps
        yield number - steps
    else:
        yield number

for i in chunk(35000):
   print(i)

# prints (wrongly):
# 10000
# 25000

或者有内置库吗？

Answer 1

我认为这可以稍微清理一下，所以请在接下来的几分钟内回来查看是否有编辑，但是:

def chunk(num, thresh):
    while True:
        if num > thresh:
            num -= thresh
            yield thresh
        else:
            yield num
            break

for x in chunk(22500, 10000):
    print(x)

输出

10000100002500

Edit:

Consider:

def chunk(num, thresh):
    while num:
        to_yield = min(num, thresh)
        yield to_yield
        num -= to_yield

for x in chunk(22500, 10000):
    print(x)