Python Graph.addEdgesFromList Examples

Example #1

File: graph_generators.py Project: pfilo8/Diffusion-Processes-on-Complex-Networks

def generate_ring_graph(n, k):
    g = Graph()
    nodes = list(range(n))
    for j in range(1, k // 2 + 1):
        targets = nodes[j:] + nodes[0:j]  # first j nodes are now last in list
        g.addEdgesFromList(zip(nodes, targets))
    return g

Example #2

File: main2.py Project: pfilo8/Diffusion-Processes-on-Complex-Networks

import os

import numpy as np
import pandas as pd

from model import Graph

DATA_DIR = 'data'
DATA_CSV = 'karate.csv'
OUTPUT_DIR = 'outputs'
OUTPUT_CSV = 'karate_club_nn.csv'

if __name__ == '__main__':
    df = pd.read_csv(os.path.join(DATA_DIR, DATA_CSV))
    edge_list = df.values

    g = Graph()
    g.addEdgesFromList(edge_list)
    results = g.getShortestPath(7)
    results = pd.DataFrame(results.items(), columns=['id', 'distance'])
    results.to_csv(os.path.join(OUTPUT_DIR, OUTPUT_CSV), index=False)