toy_webcounter_networkx_bipartite_graph.py

import networkx as nx

B = nx.Graph()
# Add nodes with the node attribute "bipartite"
B.add_nodes_from([1, 2, 3, 4], bipartite=0)
B.add_nodes_from(["a", "b", "c"], bipartite=1)
# Add edges only between nodes of opposite node sets
B.add_edges_from([(1, "a"), (1, "b"), (2, "b"), (2, "c"), (3, "c"), (4, "a")])

def get_webpage_visitors(webpage, target_graph):
    for page in target_graph.adjacency():
         if page[0] == webpage:
              return list(page[1].keys())
         else:
              return None

get_webpage_visitors(1, B)


def get_webpage_count(webpage, target_graph):
    for page in target_graph.adjacency():
        if page[0] == webpage:
            return len(page[1].keys())
        else:
            return None

get_webpage_count(1, B)


def get_visitor_list(visitor, target_graph):
    for page in target_graph.adjacency():
        if page[0] == visitor:
            return list(page[1].keys())