循环网格中的星形

Question

我正在使用 python 3， 我有一个 9 * 9 的列表或网格，左上角是 (0, 0)，右下角是 (8, 8)

我只想遍历该网格上距中心 (4, 4) 4 个或更少单元格的每个项目，

并对该单元格执行操作。

这是我当前的代码（不工作）-

for row in range(9):
    amount = row * 2 + 1
    for col in range(9):
        if amount == 0:
            continue
        print("Col- ", col + 4 -(row % 4), "Row- ", row)
        amount -= 1

Answer 1

我只成功地循环了蓝色方块，这是我的代码：

import numpy as np


def index_generator(grid_dim):
    center = grid_dim//2
    dx = center + 1
    sx = center

    while(dx > 0 and sx < grid_dim):
        dx -= 1
        sx += 1
        yield range(dx, sx)

    while(dx <= (center) and sx >= (center + 1)):
        dx += 1
        sx -= 1
        yield range(dx, sx)


grid_dim = 9

A = np.zeros((grid_dim, grid_dim))

for r, i in zip(index_generator(grid_dim), range(grid_dim)):
    for j in r:
        A[i, j] = 1

print(A)

输出是：

[[0. 0. 0. 0. 1. 0. 0. 0. 0.]
 [0. 0. 0. 1. 1. 1. 0. 0. 0.]
 [0. 0. 1. 1. 1. 1. 1. 0. 0.]
 [0. 1. 1. 1. 1. 1. 1. 1. 0.]
 [1. 1. 1. 1. 1. 1. 1. 1. 1.]
 [0. 1. 1. 1. 1. 1. 1. 1. 0.]
 [0. 0. 1. 1. 1. 1. 1. 0. 0.]
 [0. 0. 0. 1. 1. 1. 0. 0. 0.]
 [0. 0. 0. 0. 1. 0. 0. 0. 0.]]

我使用生成器来构造每一行的索引，然后循环遍历它，在第一部分中，我找到了中心并声明了两个分隔符（sx 和 dx）。比我循环移动分隔符，直到它到达矩阵的边界，然后我只是做同样的事情，但向中心移动。

希望对您有所帮助。