Python：在两个值之间生成n个随机整数，求和为给定的数字

Question

我非常想在两个值（min，max）之间生成n随机整数，其和等于给定的数字m。

注意：我在StackOverflow中发现了类似的问题；但是，它们并没有准确地解决这个问题（使用Dirichlet函数，因此数字介于0和1之间）。

例如：我需要8个0到24之间的随机数（整数），其中8个生成的数字的总和必须等于24。

感谢您的帮助。谢谢

Answer 1

这是一个基于这个答案的可能解决方案。看来dirichlet方法只在0和1之间起作用。最初的答案应该是完全正确的。一旦你评论说它达到了你的目的，我很乐意删除它。

别忘了对原来的答案投赞成票。

target = 24

x = np.random.randint(0, target, size=(8,))
while sum(x) != target: 
    x = np.random.randint(0, target, size=(8,))

print(x)
# [3 7 0 6 7 0 0 1]

Answer 2

这是分区数理论的一个例子。这是解决办法

def partition(n,k,l, m):
    if k < 1:
        raise StopIteration
    if k == 1:
        if n <= m and n>=l :
            yield (n,)
        raise StopIteration
    for i in range(l,m+1):
        for result in partition(n-i,k-1,i,m):                
            yield result+(i,)

n = 24 # sum value
k = 8 # partition size
l = 0 # range min value
m = 24 # range high value 

result = list(partition(n,k,l,m ))

这将给出满足条件的所有组合。ps这相当慢，因为它给出了分区大小的所有情况。

Answer 3

你可以使用整数分布，它自然地和一些固定数-多项式。

只要前后移动，它就会自动工作

密码

import numpy as np

def multiSum(n, p, maxv):
    while True:
        v  = np.random.multinomial(n, p, size=1)
        q  = v[0]
        a,  = np.where(q > maxv) # are there any values above max
        if len(a) == 0: # accept only samples below or equal to maxv
            return q

N = 8
S = 24
p = np.full((N), 1.0/np.float64(N))

mean  = S / N
start = 0
stop  = 24
n = N*mean - N*start

h = np.zeros((stop-start), dtype=np.int64)
print(h)
for k in range(0, 10000):
    ns = multiSum(n, p, stop-start) + start # result in [0...24]
    #print(np.sum(ns))
    for v in ns:
        h[v-start] += 1

print(h)