如何在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}

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

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))