跳转到主内容
极星编程网:以代码为星,赴技术山海!

如何在二维网格中根据线性索引精确设置指定位置为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)
✅ 关键改进说明: Python 3.14.3 微软官方的 Python 扩展,是 VS Code 安装量最高的扩展(209M+)。集成 IntelliSense(通过 Pylance)、调试(通过 Python Debugger)、代码检查、格式化、重构和单元测试等功能。支持 Jupyter Notebook、虚拟环境管理和多 Python 版本切换。 下载 divmod(i-1, size) 替代手动计算 // 和 %,简洁且高效; 边界检查 (if 0 <= row < size...)增强鲁棒性,防止非法索引崩溃; 动态格式化 (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 列表进行,无需修改核心逻辑。 这套方案兼顾正确性、可读性与可维护性,是处理网格索引映射问题的工程级参考实现。

相关文章