Python：查找所有帶有約束元素的元組，並將它們合成一個常數

Question

假設給出了一個長度爲4的矢量，其中最小值（0,0,0,0）和最大值（2,1,3,2） ）。我想找到在此範圍內，加起來，說所有的整數向量，5

下面的代碼實現了：

maxvalues = (2,1,3,2) 
k = 5 
t = [] 
for x1 in range(maxvalues[0]+1): 
    for x2 in range(maxvalues[1]+1): 
     for x3 in range(maxvalues[2]+1): 
      for x4 in range(maxvalues[3]+1): 
        if x1+x2+x3+x4 is k: 
         t.append((x1,x2,x3,x4)) 

但是，此代碼不能一概而論很好地大載體和似乎有點低效。有沒有人有一個很好的方式來概括這個？

編輯：稍有不慎，忘了if語句

Answer 1

使用itertools.product

from itertools import product 

for x1, x2, x3, x4 in product(*(range(i+1) for i in maxvalues)): 
    t.append((x1, x2, x3, x4)) 

雖然更好（更普遍）將在product對象直接轉換成一個列表。

t = list(product(*(range(i+1) for i in maxvalues))) 

隨着if聲明包含，它會是這個樣子。

t = [i for i in product(*(range(i+1) for i in maxvalues)) if sum(i) == k]