【问题标题】:Remove diagonals from a grid从网格中删除对角线
【发布时间】:2018-09-27 10:48:13
【问题描述】:

我正在尝试在 python 中创建一个高度为 5 和宽度为 10 的图形。到目前为止,我有这段代码构建了一个图形,该图形具有对角线的边以及图形中的上、左、右和下边:

def construct_graph(data):
# initialize all of the astar nodes
nodes = [[ANode(x, y) for y in range(data['height'])] for x in range(data['width'])]
graph = {}
# make a graph with child nodes
for x, y in product(range(data['width']), range(data['height'])):
    node = nodes[x][y]
    graph[node] = []
    for i, j in product([-1, 0, 1], [-1, 0, 1]):
        if not (0 <= x + i < data['width']): continue
        if not (0 <= y + j < data['height']): continue
        if [x+i,y+j] in data['obstacle']: continue
        graph[nodes[x][y]].append(nodes[x+i][y+j])
return graph, nodes

如何修改上述函数以仅创建上、左、右和下链接?

注意:ANode 只是一个普通的 python 类,它为一个节点存储x, y

【问题讨论】:

    标签: arrays algorithm python-2.7 matrix graph-algorithm


    【解决方案1】:

    不要将product 用于内部循环,只需指定您想要的邻居:

    def construct_graph(data):
    # initialize all of the astar nodes
    nodes = [[ANode(x, y) for y in range(data['height'])] for x in range(data['width'])]
    graph = {}
    # make a graph with child nodes
    for x, y in product(range(data['width']), range(data['height'])):
        node = nodes[x][y]
        graph[node] = []
        for i, j in [(-1, 0), (0, -1), (0, 1), (1, 0)]:
            if not (0 <= x + i < data['width']): continue
            if not (0 <= y + j < data['height']): continue
            if [x+i,y+j] in data['obstacle']: continue
            graph[nodes[x][y]].append(nodes[x+i][y+j])
    return graph, nodes
    

    【讨论】: