このアルゴリズムを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に置き換えました。

これを明示的に書くにはどうすればよいですか？より具体的には、kが、p = 0からNまでの{p * K、...、（p + 1）* K-1}の集合にあれば、何かを行う。コードでどうすればいいですか？

Answer 1

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

Answer 2

あなたは3つのループを持つことができますか？

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