def makeGexf(): raw_file_name = "gexf_format_test.csv" #raw_file_name = "HKALL_tags_RT10count.csv" gexf = Gexf("Ching-Ya Lin", "Keyword relation by hours.") graph = gexf.addGraph("undirected", "dynamic", "graph!", "datetime") wid = graph.addNodeAttribute("Weight", "1", "integer", "dynamic") ewid = graph.addEdgeAttribute("EdgeWeight", "1", "integer", "dynamic") if (1): f = codecs.open(raw_file_name, "r", "utf-8") #with open(raw_file_name, 'r') as f: reader = csv.reader(f) w_d = defaultdict() #weight_dictionary for all keywords e_w_d = defaultdict() #edge for row in reader: #row = [x.decode('utf8') for x in rowx] tt = time.strptime(row[0], "%Y-%m-%d %H") rtt = time.strftime("%Y-%m-%d %H:00:00", tt) words = row[1].split('/') templist = [] #one row list to help addEdge for w in words: # useless : w = w.encode('utf8') #print w #print chardet.detect(w) templist.append(w) if graph.nodeExists(w): n = graph.nodes[w] v = w_d[w] + 1 #v = int(n.attributes[int(wid)]["value"]) #n.attributes will get the initial value which we don't want. n.addAttribute(wid, str(v), rtt) w_d[w] = v else: new = graph.addNode(w, w, rtt) new.addAttribute(wid, "1", rtt) w_d[w] = 1 for w1 in templist: for w2 in templist: if w1 != w2: s1 = w1 + w2 s2 = w2 + w1 if graph.edgeExists(s1): e = graph.edges[s1] val = e_w_d[s1] + 1 e.addAttribute(ewid, str(val), rtt) e_w_d[s1] = val elif graph.edgeExists(s2): e = graph.edges[s2] val = e_w_d[s2] + 1 e.addAttribute(ewid, str(val), rtt) e_w_d[s2] = val else: enew = graph.addEdge(s1, w1, w2, 1, rtt) enew.addAttribute(ewid, "1", rtt) e_w_d[s1] = 1 templist.remove(w1) output_file = open("eventhr.gexf", "w") gexf.write(output_file)
for bankID in bankList: graph.nodes[bankID].addAttribute('state',value = bankList[bankID][1], start =str(bankList[bankID][0]), end=str(EndTime)) for bankID in diff(range(len(graph.nodes.keys())),bankList.keys()): graph.nodes[bankID].addAttribute('state',value = 1, start ='0', end=str(EndTime)) return graph # Creating toy example # state = 1 => solvant # state = 2 => exposed # state = 3 => feail # state = 4 => dead List = [] # It is important that the list is ordered by time List.append((1,0,1)) #(time, bankID, state) List.append((1,1,4)) List.append((2,0,4)) List.append((3,2,2)) List.append((4,2,3)) List.append((5,2,4)) # Creating list of edges to be cut at a specific time EdgeList = {} EdgeList[(2,20)]=3 EdgeList[(2,21)]=4 EdgeList[(2,6)]=5 gexf = Gexf("Financial Contagion","A dynamic network model") G0 = pickle.load(open('/Users/linnlii/Documents/GitHub/pygexf/test/ToyGraph.p', 'rb')) graph = convertGraphs(G0,EdgeList) graph = addAttributes(graph,List) output_file=open("DynamicNetworks.gexf","w") gexf.write(output_file)
import sys, pprint sys.path.append('../gexf') from _gexf import Gexf, GexfImport # test helloworld.gexf gexf = Gexf("Paul Girard", "A hello world! file") graph = gexf.addGraph("directed", "static", "a hello world graph") graph.addNode("0", "hello") graph.addNode("1", "World") graph.addEdge("0", "0", "1") output_file = open("helloworld.gexf", "w") gexf.write(output_file) # test GexfImport f = open("gexf.net.dynamics_openintervals.gexf") gexf_import = Gexf.importXML(f) f.close() f = open("gexf.net.dynamics_openintervals.gexf") gexf_import2 = Gexf.importXML(f) f.close() print "test gexf comparision " + str(gexf_import == gexf_import2) graph = gexf_import.graphs[0] # display nodes list for node_id, node in graph.nodes.iteritems(): print node.label pprint.pprint(node.getAttributes(), indent=1, width=1)