python读取txt文件存为邻接矩阵

python读取txt文件存为邻接矩阵

import numpy as np
def getdata(filename):

    linedata = open("socfb-Amherst41.txt", 'r')    #读取txt文件
    cnt = 0
    res = []        #存读取后的数据，二元列表
    n = 2235
    matrix = np.zeros((n, n))
    for line in linedata:
        linelist = [int(s) for s in line.split()] #每一行根据分割后的结果存入列表
        res.append([])
        for x in linelist:
            res[cnt].append(x)
            matrix[max(res[cnt]) - 1][min(res[cnt]) - 1] = 1
            matrix[min(res[cnt]) - 1][max(res[cnt]) - 1] = 1
        cnt += 1
    print(matrix)
    return res

filename = 'socfb-Amherst41.txt'
data = getdata(filename)

输出结果就是一个邻接矩阵