Как написать этот алгоритм в коде 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 * K, ..., (p + 1) * K-1} для p = 0 до N, то сделайте что-нибудь. Как я могу сделать это в коде?

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

У вас может быть только три петли, нет?

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