例:随机生成一个长度为1000, 总和为2021的列表。
import numpy as np
def allocation_amount(num_people, amount):
a = [np.random.randint(0, amount) for i in range(num_people-1)]
a.append(0)
a.append(amount)
a.sort()
b = [a[i+1]-a[i] for i in range(num_people)]
b = np.array(b)
return b
bonus = allocation_amount(1000, 2021)
print(bonus)
print(np.sum(bonus))