networkx.grid_2d_graph

grid_2d_graph(m, n, periodic=False)

Return the 2d grid graph of mxn nodes, each connected to its nearest neighbors. Optional argument periodic=True will connect boundary nodes via periodic boundary conditions.