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

bravozyz

于 2021-05-16 00:39:02 发布

阅读量419

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签： python

本文链接：https://blog.csdn.net/weixin_41008021/article/details/116862516

算法专栏收录该内容

3 篇文章

订阅专栏

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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