smMeasure/visual.py

58 lines
1.3 KiB
Python
Raw Permalink Normal View History

2017-10-15 19:59:50 +00:00
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Mon Aug 7 09:44:33 2017
@author: eddie
"""
import numpy as np
import networkx as nx
#import matplotlib.pyplot as plt
#import ast
#import json
SymMat = open("SymMat.txt",'r')
GrpSz = open("GrpSz.txt",'r')
Mat = SymMat.read()
sz = int(GrpSz.read())
Mat = Mat.replace(" ","")
Mat = Mat.replace("\n","")
Mat = Mat.replace("[","")
Mat = Mat.replace("]","")
Mat = Mat.split(",")
Mat = [float(i) for i in Mat]
Mat = [Mat[x:x+sz] for x in range(0, len(Mat), sz)]
Mat = np.array(Mat)
Mat = np.triu(Mat)
#Mat = json.loads(Mat)
#Mat = ast.literal_eval(Mat)
graph = nx.from_numpy_matrix(Mat)
graph.edges(data=True)
#layout = nx.circular_layout(graph) # or whatever layout you wish
#nx.draw_circular(graph)
#mapping = {0:r'$e$',1:r'$\delta$',2:r'$\sigma$',3:r'$\rho$',4:r'$\rho^2$',5:r'$\gamma$'}
#graph=nx.relabel_nodes(graph,mapping)
"""
edgeLabels = {} # dictionary of node tuples to edge labels: {(nodeX, nodeY): aString}
for a, b in graph.edges(): # loop over all the edges
edgeLabels[(a, b)] = str(graph.get_edge_data(a, b, {"weight":0})["weight"])
#nx.draw_networkx_edge_labels(graph,pos=layout, edge_labels=edgeLabels)
"""
#print(graph.edges())
#nx.draw(graph,with_labels=True)
nx.draw(graph, node_size=0.8)