问题描述
我想应用从文件文本中获取数据的相邻矩阵并应用矩阵然后输出将是零和一
—
解决方法
我完全假设了一切,希望这就是您要找的。下面的代码,将设置邻接矩阵(首先只包含0),然后将相邻节点的0转换为1。
class Graph:
def __init__(self,numNodes):
self.adjacencyMatrix = []
for i in range(numNodes):
self.adjacencyMatrix.append([0 for i in range(numNodes)])
self.numNodes = numNodes
def addEdge(self,start,end):
self.adjacencyMatrix[start][end] = 1
# random example
edges_list = [[1,2],[0,1],[2,3]]
# assuming number of nodes is 4
graph = Graph(4)
# converts 0's to 1's for adjacent nodes
for i in edges_list:
graph.addEdge(i[0],i[1])
with open("codon2.txt","w") as f_out:
for i in graph.adjacencyMatrix:
for j in i:
f_out.write(str(j))
f_out.write('\n')