生成随机数以获得固定的总和（python）

Question

我有以下清单：

Sum=[54,1536,36,14,9,360]

我需要生成4个其他列表，其中每个列表将包含从0开始的6个随机数，并且这些数字将加起来为总和。例如

l1=[a,b,c,d,e,f] where a+b+c+d+e+f=54
l2=[g,h,i,j,k,l] where g+h+i+j+k+l=1536

直到l6。我需要在python中执行此操作。能做到吗

Answer 1

这可能不是最有效的方法，但是会起作用

totals = [54, 1536, 36, 14]

nums = []
x = np.random.randint(0, i, size=(6,))
for i in totals:
    while sum(x) != i: x = np.random.randint(0, i, size=(6,))
    nums.append(x)
print(nums)

[array（[3，19，21，11，0，0]），array（[111，155，224，511，457，78]），array（[8，5，4，12，2，5] ），array（[3，1，3，2，1，4]）]

这是一种更有效的方法

总计= [54,1536,36,14,9,360，0]

nums = []
for i in totals:
    if i == 0: 
        nums.append([0 for i in range(6)])
        continue
    total = i
    temp = []
    for i in range(5):
        val = np.random.randint(0, total)
        temp.append(val)
        total -= val
    temp.append(total)
    nums.append(temp)

print(nums)

[[22，4，16，0，2，10]，[775，49，255，112，185，160]，[2，10，18，2，0，4]，[10，2，1， 0，0，1]，[8，0，0，0，0，1]，[330，26，1，0，2，1]，[0，0，0，0，0，0]]

Answer 2

生成总和为某个整数的随机数列表是一项非常艰巨的任务。跟踪剩余数量并根据剩余可用数量顺序生成项目会导致分布不均匀，其中系列中的第一个数字通常比其他数字大得多。最重要的是，最后一项将始终与零不同，因为列表中的前一项总和永远不会达到所需的总数（随机生成器通常使用最大的开放间隔）。一代又一代地改组列表可能会有所帮助，但通常也不会给出好的结果。

一个解决方案可能是生成随机数，然后对结果进行标准化，如果您需要它们是整数，则最终将其四舍五入。

import numpy as np
totals = np.array([54,1536,36,14])  # don't use Sum because sum is a reserved keyword and it's confusing

a = np.random.random((6, 4))  # create random numbers
a = a/np.sum(a, axis=0) * totals  # force them to sum to totals

# Ignore the following if you don't need integers
a = np.round(a)  # transform them into integers
remainings = totals - np.sum(a, axis=0)  # check if there are corrections to be done
for j, r in enumerate(remainings):  # implement the correction
    step = 1 if r > 0 else -1
    while r != 0:
        i = np.random.randint(6)
        if a[i,j] + step >= 0:
            a[i, j] += step
            r -= step

的每一列a代表您想要的列表之一。希望这可以帮助。