python - 如何从python中的类访问字典值？

Question

我正在尝试使用类来使用以下代码创建字典：

import numpy as np
import collections


class Graph:
    def __init__(self):
        self.graph = collections.defaultdict(dict)

    def add_edge(self, u, v, weight = 1, directed = True):
        self.graph[u][v] = weight
        if not directed:
            self.graph[v][u] = weight

    def __str__(self):
        to_return = ''
        for vertex in self.graph:
            to_return += str(vertex) + ': '
            for edge in self.graph[vertex]:
                to_return +=  '(' + str(edge) + ', ' + str(self.graph[vertex][edge]) + ')'
                to_return += '   '

            to_return += '\n'
        return to_return

link_i = [1, 1, 3, 3, 3, 4, 4, 5, 5, 6]
link_j = [2, 3, 1, 2, 5, 5, 6, 6, 4, 4]

if __name__ == '__main__':
    g = Graph()
    for i in range(len(link_i)):
        g.add_edge(link_i[i],link_j[i])

    print(g)
    
for key in g.graph:
    print(g.graph[key])
 

这给出了一个输出是字典

{2: 1, 3: 1}
{1: 1, 2: 1, 5: 1}
{5: 1, 6: 1}
{6: 1, 4: 1}
{4: 1}

如何将输出作为数组列表而不是字典列表？

标签： pythonclassdictionarykey

Answer 1

您可以为您的类提供方法来包装此功能：

class Graph:
    # all the other stuff

    def verteces(self):
        return self.graph.keys()

    def neigbours(self, vertex):
        return list(self.graph[vertex].keys())

g = Graph()
# fill the graph

for v in g.verteces():
    print(g.neighbours(v))