我如何从 python 中的文本文件读取我的数据集并在其上应用矩阵

问题描述

我想应用从文件文本中获取数据的相邻矩阵并应用矩阵然后输出将是零和一

这是我存储在文件文本中的数据集的代码

解决方法

我完全假设了一切,希望这就是您要找的。下面的代码,将设置邻接矩阵(首先只包含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')