Как написать этот алгоритм в коде python?

У меня есть следующий код.

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, то сделайте что-нибудь. Как я могу сделать это в коде?