在Python中使用常数和迭代索引

我正在使用多项式，并且必须执行一些操作，其中我的循环变量的总和低于某些常数d。现在我有在Python中使用常数和迭代索引

for i in range(0, d): 
    for j in range(i, d): 
     for k in range(j, d):

这似乎有点难看给我。是否有一些函数，大概在itertools，允许我迭代for i, j, k in foo(d)？

来源

2016-11-21 P. Camilleri

给出的范围之间的依赖关系，我不觉得有什么具体到什么帮助。 – AChampion

你可以自己写。这里有一个强力的方式为3个变量：

def constant_sum(s): 
    for i in range(s+1): 
     for j in range(s-i+1): 
      k = s - i - j 
      yield i,j,k 

def inferior_sum(s): 
    for i in range(s+1): 
     for j in range(s+1): 
      if i+j >= s: 
       break 
      for k in range(s+1): 
       if i+j+k < s: 
        yield i,j,k 
       else: 
        break 

for i,j,k in constant_sum(3): 
    print(i,j,k) 
print() 
for i,j,k in inferior_sum(3): 
    print(i,j,k)

输出：

下面是递归的版本，可以为任何款项（S）做任何数量的变量（N）...轻轻测试：

def constant_sum(n,s): 
    if n == 1: 
     yield [s] 
    else: 
     for i in range(s+1): 
      for r in constant_sum(n-1,s-i): 
       yield [i] + r 

def inferior_sum(n,s): 
    if n == 1: 
     for i in range(s): 
      yield [i] 
    else: 
     for i in range(s): 
      for r in inferior_sum(n-1,s-i): 
       yield [i] + r 

for x in constant_sum(3,3): 
    print(*x) 
print() 
for x in inferior_sum(3,3): 
    print(*x)

输出：

来源

2016-11-21 20:30:22

+1我真的很喜欢递归方法。你可以用元组来做到这一点，以保持用户友好的界面（也稍微快一点）。 – stefan

迭代工具/功能的方法是这样的：

from itertools import product 
inferior_sum3 = filter(lambda x: sum(x)<3, product(range(4),range(4),range(4))) 
for permu in inferior_sum3: 
    print(permu)

输出：

(0, 0, 0) 
(0, 0, 1) 
(0, 0, 2) 
(0, 1, 0) 
(0, 1, 1) 
(0, 2, 0) 
(1, 0, 0) 
(1, 0, 1) 
(1, 1, 0) 
(2, 0, 0)

来源

2016-11-21 20:45:35 Ewoud

在Python中使用常数和迭代索引

回答

相关问题