python - 如何在 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 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