使用权重打印邻接矩阵

时间:2018-03-02 20:00:53

标签: python-2.7 adjacency-matrix

我想知道我是否可以获得一些帮助来打印python中的邻接矩阵。我希望获得的输出如下:

![enter image description here

代码如下:

import numpy

adjlist = [
          (1,2,15),
          (1,4,7),
          (1,5,10),
          (2,3,9),
          (2,4,11),
          (2,6,9),
          (3,5,12),
          (3,6,7),
          (4,5,8),
          (4,6,14),
          (5,6,8)
         ]


def matfn(adjlist, nodes):
    '''Returns a (weighted) adjacency matrix as a NumPy array.'''
    matrix = []

    for node in nodes:
        weights = {endnode:int(weight)
                   for w in adjlist.get(node, {})
                   for endnode, weight in w.items()}
        matrix.append([weights.get(endnode, 0) for endnode in nodes])
        matrix = numpy.array(matrix)
        return matrix + matrix.transpose()

matfn(adjlist, nodes=list('123456'))

(对python来说很新......) 提前谢谢......

0 个答案:

没有答案