• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• cdlib-master
  • .github
    • PULL_REQUEST_TEMPLATE.md
    • ISSUE_TEMPLATE
      • feature_request.md
      • bug_report.md
    • CONTRIBUTING.md
    • CODE_OF_CONDUCT.md
  • _config.yml
  • .coveragerc
  • requirements.docs.txt
  • cdlib
    • ensemble
      • bunch_executions.py
      • __init__.py
    • classes
      • edge_clustering.py
      • __init__.py
      • fuzzy_node_clustering.py
      • attr_node_clustering.py
      • bipartite_node_clustering.py
      • clustering.py
      • node_clustering.py
    • viz
      • __init__.py
      • networks.py
      • plots.py
    • test
      • test_partitions_comparisons.py
      • test_utils.py
      • test_fitness_functions.py
      • test_io.py
      • test_bunch_executions.py
      • test_edgeclustering.py
      • test_nodeclustering.py
      • __init__.py
      • test_community_discovery_models.py
      • test_viz_plots.py
      • test_ranking.py
      • test_attributeclustering.py
      • test_viz_network.py
    • algorithms
      • overlapping_partition.py
      • edge_clustering.py
      • attribute_clustering.py
      • crisp_partition.py
      • internal
        • LEMON.py
        • Markov.py
        • LAIS2_nx.py
        • FuzzyCom.py
        • DER.py
        • AGDL.py
        • SLPA_nx.py
        • ILouvain.py
        • em.py
        • lfm.py
        • HLC.py
        • GDMP2_nx.py
        • multicom.py
        • scan.py
        • NodePerception.py
        • __init__.py
        • CONGO.py
        • CONGA.py
        • OSSE.py
        • BIGCLAM.py
        • PercoMCV.py
      • __init__.py
      • biparitte_clustering.py
    • readwrite
      • io.py
      • __init__.py
    • evaluation
      • fitnessranking.py
      • internal
        • TOPSIS.py
        • link_modularity.py
        • __init__.py
        • onmi.py
        • statistical_ranking.py
      • __init__.py
      • comparisonranking.py
      • comparison.py
      • fitness.py
    • __init__.py
    • utils.py
  • LICENSE
  • requirements_optional.txt
  • setup.py
  • setup.cfg
  • .travis.yml
  • README.md
  • requirements.txt
  • .gitignore
  • docs
    • download.rst
    • Makefile
    • make.bat
    • installing.rst
    • tutorial.rst
    • overview.rst
    • bibliography.rst
    • CDlib.ipynb
    • reference
      • classes.rst
      • classes
        • node_clustering.rst
        • fuzzy_node_clustering.rst
        • bi_node_clustering.rst
        • edge_clustering.rst
        • attr_node_clustering.rst
      • reference.rst
      • algorithms.rst
      • viz.rst
      • ensemble.rst
      • evaluation.rst
      • cd_algorithms
        • node_clustering.rst
        • edge_clustering.rst
      • utils.rst
      • readwrite.rst
    • developer
      • developer.rst
    • index.rst
    • conf.py
    • _static
      • css
        • custom.css
  • .pyup.yml

from cdlib import NodeClustering, FuzzyNodeClustering, EdgeClustering
import json

__all__ = ["write_community_csv", "read_community_csv", "write_community_json",
           "read_community_json", "read_community_from_json_string"]


def write_community_csv(communities,  path, delimiter=","):
    """
    Save community structure to comma separated value (csv) file.

    :param communities: a NodeClustering object
    :param path: output filename
    :param delimiter: column delimiter

    :Example:

    >>> import networkx as nx
    >>> from cdlib import algorithms, readwrite
    >>> g = nx.karate_club_graph()
    >>> coms = algorithms.louvain(g)
    >>> readwrite.write_community_csv(coms, "communities.csv", ",")

    """
    with open(path, "w") as f:
        for cid, community in enumerate(communities.communities):
            res = delimiter.join(list(map(str, community)))
            f.write("%s\n" % res)


def read_community_csv(path, delimiter=",", nodetype=str):
    """
    Read community list from comma separated value (csv) file.

    :param path: input filename
    :param delimiter: column delimiter
    :param nodetype: specify the type of node labels, default str
    :return: NodeClustering object

    :Example:

    >>> import networkx as nx
    >>> from cdlib import algorithms, readwrite
    >>> g = nx.karate_club_graph()
    >>> coms = algorithms.louvain(g)
    >>> readwrite.write_community_csv(coms, "communities.csv", ",")
    >>> coms = readwrite.read_community_csv(coms, "communities.csv", ",", str)

    """
    communities = []
    with open(path) as f:
        for row in f:
            community = list(map(nodetype, row.rstrip().split(delimiter)))
            communities.append(list(community))

    return NodeClustering(communities, None, "")


def write_community_json(communities, path):
    """
    Generate a JSON representation of the clustering object

    :param communities: a cdlib clustering object
    :param path: output filename
    :return: a JSON formatted string representing the object

    :Example:

    >>> import networkx as nx
    >>> from cdlib import algorithms, readwrite
    >>> g = nx.karate_club_graph()
    >>> coms = algorithms.louvain(g)
    >>> readwrite.write_community_json(coms, "communities.json")
    """

    partition = {"communities": communities.communities, "algorithm": communities.method_name,
                 "params": communities.method_parameters, "overlap": communities.overlap, "coverage": communities.node_coverage}

    try:
        partition['allocation_matrix'] = communities.allocation_matrix
    except AttributeError:
        pass

    js_dmp = json.dumps(partition)
    with open(path, "w") as f:
        f.write(js_dmp)


def read_community_json(path):
    """
    Read community list from JSON file.

    :param path: input filename
    :return: a Clustering object

    :Example:

    >>> import networkx as nx
    >>> from cdlib import algorithms, readwrite
    >>> g = nx.karate_club_graph()
    >>> coms = algorithms.louvain(g)
    >>> readwrite.write_community_json(coms, "communities.json")
    >>> readwrite.read_community_json(coms, "communities.json")
    """

    with open(path, "r") as f:
        coms = json.load(f)

    nc = NodeClustering([list(c) for c in coms['communities']], None, coms['algorithm'],
                        coms['params'], coms['overlap'])
    nc.node_coverage = coms['coverage']

    if 'allocation_matrix' in coms:
        nc.__class__ = FuzzyNodeClustering
        nc.allocation_matrix = coms['allocation_matrix']

    if type(nc.communities[0][0]) is list:
        cms = []
        for c in nc.communities:
            cm = []
            for e in c:
                cm.append(tuple(e))
            cms.append(list(cm))
        nc.communities = cms
        nc.__class__ = EdgeClustering

    return nc


def read_community_from_json_string(json_repr):
    """
    Read community list from JSON file.

    :param json_repr: json community representation
    :return: a Clustering object

    :Example:

    >>> import networkx as nx
    >>> from cdlib import algorithms, readwrite
    >>> g = nx.karate_club_graph()
    >>> coms = algorithms.louvain(g)
    >>> readwrite.write_community_json(coms, "communities.json")
    >>> with open("community.json") as f:
    >>>     cr = f.read()
    >>>     readwrite.write_community_from_json_string(cr)
    """

    coms = json.loads(json_repr)

    nc = NodeClustering([list(c) for c in coms['communities']], None, coms['algorithm'],
                        coms['params'], coms['overlap'])
    nc.node_coverage = coms['coverage']

    if 'allocation_matrix' in coms:
        nc.__class__ = FuzzyNodeClustering
        nc.allocation_matrix = coms['allocation_matrix']

    if type(nc.communities[0][0]) is list:
        cms = []
        for c in nc.communities:
            cm = []
            for e in c:
                cm.append(tuple(e))
            cms.append(tuple(cm))
        nc.communities = cms
        nc.__class__ = EdgeClustering

    return nc