如何在二维网格中根据线性索引精确设置指定位置为1

本文详解如何将一维编号（如1~16）映射到二维网格坐标，并动态生成含指定“激活位”的二进制矩阵，同时提供健壮、可扩展的打印格式，避免索引错位与对齐混乱。

在网格编程中，一个常见需求是：给定网格尺寸 size 和一组从 1 开始的线性索引（如 [2, 5, 12]），将其准确转换为二维坐标 (row, col) 并在对应位置填入 1，其余保持 0。关键在于理解索引的行优先（row-major）布局规则：索引 i 对应的位置满足：

• 行号 row = (i - 1) // size
• 列号 col = (i - 1) % size

这是因为索引 1 对应 (0, 0)（首行首列），2 → (0, 1)，…，size+1 → (1, 0)，依此类推。

以下是一个结构清晰、符合 Python 最佳实践（PEP 8、关注点分离）的完整实现：

import random

def make_grid(size, chosen_indices):
    """创建 size×size 二进制网格，将 chosen_indices 中的线性索引位置设为 1"""
    grid = [[0] * size for _ in range(size)]
    for i in chosen_indices:
        row, col = divmod(i - 1, size)  # 等价于 ((i-1)//size, (i-1)%size)
        if 0 <= row < size and 0 <= col < size:  # 边界防护（可选但推荐）
            grid[row][col] = 1
    return grid

def print_grid(grid):
    """美观打印网格，支持任意尺寸（自动对齐行列索引）"""
    if not grid:
        return
    size = len(grid)
    max_idx_width = len(str(size - 1))  # 最大行列索引的字符宽度
    idx_fmt = f"{{:>{max_idx_width}d}}"  # 右对齐索引格式
    elem_fmt = f"{{:>{max_idx_width}d}}"  # 元素对齐宽度与索引一致

    # 打印每行：索引 + [元素]
    for row_idx, row in enumerate(grid):
        print(idx_fmt.format(row_idx), end="[")
        print(" ".join(elem_fmt.format(x) for x in row), end="]\n")

    # 打印列索引（底部对齐，前导空格与行索引宽度匹配）
    print(" " * max_idx_width, end=" ")
    print(" ".join(idx_fmt.format(col_idx) for col_idx in range(size)))

# 主程序逻辑（无全局变量，职责明确）
if __name__ == "__main__":
    size = int(input("Size? (even) "))
    area = size * size
    half = area // 2
    thechosenhalf = random.sample(range(1, area + 1), half)

    grid = make_grid(size, thechosenhalf)
    print_grid(grid)

✅ 关键改进说明：

• divmod(i-1, size) 替代手动计算 // 和 %，简洁且高效；
• 边界检查（if 0
• 动态格式化（max_idx_width）确保 size=15 或更大时行列索引与元素严格对齐；
• 函数职责单一：make_grid 负责数据构造，print_grid 仅负责输出，便于测试与复用；
• 无全局变量：所有状态通过参数传递，避免隐式依赖和调试陷阱。

? 使用示例：
当 size=4 且 thechosenhalf = [12, 5, 16, 2, 8, 9, 15, 3] 时，输出为：

0[0 1 1 0]
1[1 0 0 1]
2[1 0 0 1]
3[0 0 1 1]
  0 1 2 3 

完美匹配预期——每个数字均按 1→(0,0), 2→(0,1), ..., 5→(1,0), 12→(2,3), 16→(3,3) 规则定位。

? 注意事项：

• 输入索引必须在 1 到 size² 范围内（含），否则需额外校验；
• 若需支持“列优先”布局，只需将 divmod 改为 divmod(i-1, size) 后交换 row/col 即可；
• 后续扩展（如高亮显示、导出 CSV、可视化）可直接基于返回的 grid 列表进行，无需修改核心逻辑。

这套方案兼顾正确性、可读性与可维护性，是处理网格索引映射问题的工程级参考实现。