【Python】随机生成总和为指定整数的整数列表

最新推荐文章于 2024-04-30 20:07:38 发布

原创最新推荐文章于 2024-04-30 20:07:38 发布 · 470 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

算法专栏收录该内容

3 篇文章

订阅专栏

该博客展示了一个Python函数，利用numpy库生成一个包含特定人数且总和固定的随机数列表。通过调用`allocation_amount`函数，创建了一个长度为1000，总和为2021的列表，并确保列表元素之和等于给定的总和。最后，输出了生成的奖金分配列表及其求和结果，展示了numpy在数值计算中的应用。

例：随机生成一个长度为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))