analyse_content.py 文件源码

def sub_graph(edges,node_main,node_ip,node_cname):
    """
    ???????
    :param edges:
    :return:
    """
    from collections import Counter
    sub_graph_set = Counter()  # ?????????????
    sub_graph_count = 0    # ???????????
    sub_graph_domain_count = Counter()    #  ??????????????
    sub_graph_cname_count = Counter()
    sub_graph_ip_count = Counter()
    G = nx.Graph()
    G.add_edges_from(edges)
    C = nx.connected_component_subgraphs(G)  # ??????
    test = {'domain_count': [],
            'cname_count': [],
            'ip_count': []
            }
    for g in C:
        sub_graph_count += 1
        sub_graph_set[len(g.nodes())] += 1
        # print  list(set(g.nodes()).intersection(set(nodes_main)))
        test['domain_count'].append(len(list(set(g.nodes()).intersection(set(node_main)))))
        test['cname_count'].append(len(list(set(g.nodes()).intersection(set(node_cname)))))
        test['ip_count'].append(len(list(set(g.nodes()).intersection(set(node_ip)))))
        sub_graph_domain_count[len(list(set(g.nodes()).intersection(set(node_main))))] += 1
        sub_graph_cname_count[len(list(set(g.nodes()).intersection(set(node_cname))))] += 1
        sub_graph_ip_count[len(list(set(g.nodes()).intersection(set(node_ip))))] += 1
    # print sub_graph_set
    # print sub_graph_domain_count
    # print sub_graph_cname_count
    # print sub_graph_ip_count
    draw_graph(test)
    return sub_graph_count