python 用于2D阵列的从左到右从上到下遍历的指数生成器

Posted 2021-05-09

def diagonal(rows, cols, start = 0):
  for d in xrange(start, rows + cols - 1):
    for r in xrange(max(d - cols + 1, 0), min(d + 1, rows)):
      yield r, d - r