如何在python代码中编写此算法？

Question

我有以下代码.

for k in range( ( N + 1 ) * K ):
    if k >= 0 and k <= K-1:
        # do something
        # n = 0
    elif k >= K and k <= 2*K-1:
        # do something
        # n = 1
    elif k >= 2*K and k <= 3*K-1:
        # do something
        # n = 2
    ...
    ...

"做某事"很难解释,但我用矫揉造作n = p取而代之.

我怎么能明确写出来？

更具体地说,如果对于p = 0到N,k在集合{p*K,...,(p + 1)*K-1}中,则执行某些操作.我怎么能在代码中做到这一点？

Answer 1

你可以只有三个循环,不是吗？

for k in range(K):
  # do something
for k in range(K, 2*K-1):
  # do something
for k in range(2*K-1, (N+1)*K):
  # do the rest

Answer 2

for loop_id in xrange(N):
    for i in xrange(K):
       k = K * loop_id + i
       do_smth(loop_id, i, k)