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)
输出结果就是一个邻接矩阵