首页 > 解决方案 > 图的相邻矩阵在python中的XLS文件中

问题描述

我是 python 新手,我想制作一个相邻矩阵以将其用作图形。我得到了一个包含顶点和边的文件 .XLS。你可以在上面看到它 pinture 在此处输入图像描述

import xlrd
workbook = xlrd.open_workbook("Graph_data.XLS","rb")
sheet = workbook.sheet_by_index(0)
rows = []
for i in range(sheet.nrows):
    columns = []
    for j in range(sheet.ncols):
        columns.append(sheet.cell(i, j).value)
    rows.append(columns)

rows

标签: python-3.x

解决方案


import xlrd
workbook = xlrd.open_workbook("Graph_data.XLS","rb")
sheet = workbook.sheet_by_index(0)
l1 = []
l2 = []
for i in range(sheet.nrows):
    columns = []
    for j in range(sheet.ncols):
        columns.append(sheet.cell(i, j).value)
    l1.append(columns[0])
    l2.append(columns[1])

import numpy
l1 = numpy.asarray(l1, dtype=int)
l2 = numpy.asarray(l2,dtype=int)
uniques = list(numpy.unique(l1))+list(numpy.unique(l2))

harlinton = {}
for i in numpy.unique(uniques):
    a =list(l2[numpy.where(l1 == i)[0]])
    #b =list(l1[numpy.where(l2 == i)[0]])
    c = list(numpy.unique(a))
    harlinton[i] = c
    
harlinton


推荐阅读