egocentric_network_1_5.py 文件源码

def EgocentricNetwork(G,v):

    egocentric_network_edge_list = []
    egocentric_network_node_list = [v]
    for i in G.neighbors(v):
        egocentric_network_node_list.append(i)
        egocentric_network_edge_list.append((v,i))
    egocentric_network_node_list.sort()
    egocentric_network_edge_list = list(tuple(sorted(p)) for p in egocentric_network_edge_list)

    for i in list(itertools.combinations(egocentric_network_node_list, 2)): #generates all possible pairs of nodes
        if i in G.edges() and i not in egocentric_network_edge_list:
            egocentric_network_edge_list.append(i)

    return egocentric_network_edge_list,egocentric_network_node_list



#takes input from the file and creates a graph