Add ability to generate graphs based on correlations of sequences #25933

src/sage/graphs/generators/basic.py

@@ -20,6 +20,8 @@
 # import from Sage library
 from sage.graphs.graph import Graph
 from math import sin, cos, pi
+from numpy import corrcoef
+from sage.all import Matrix
 
 
 def BullGraph():
@@ -393,6 +395,36 @@ def CompleteGraph(n):
     G.add_edges(((i, j) for i in range(n) for j in range(i + 1, n)))
     return G
 
+def CorrelationGraph(seqs, alpha, include_anticorrelation):
+    """
+    Constructs and returns a correlation graph with a node corresponding to each sequence in `seqs`.
+
+    Edges are added between nodes where the corresponding sequences have a correlation coeffecient greater than alpha.
+
+    If include_anticorrelation is true, then edges are also added between nodes with correlation coeffecient less than -alpha.
+
+    EXAMPLES:
+
+        sage: from sage.graphs.correlation_graph import CorrelationGraph
+        sage: data=[[1,2,3],[4,5,6],[7,8,9999]]
+        sage: CG = CorrelationGraph(data, 0.9, False)
+        sage: CG
+        Looped graph on 3 vertices
+    """
+
+    # compute pairwise correlation coeffecients
+    corrs = corrcoef(seqs)
+
+    # compare against alpha to get adjacency matrix
+    if include_anticorrelation:
+        boolean_adjacency_matrix = abs(corrs)>=alpha
+    else:
+        boolean_adjacency_matrix = corrs>=alpha
+
+    adjacency_matrix = Matrix(boolean_adjacency_matrix.astype(int))
+
+    # call graph constructor
+    return Graph(adjacency_matrix, format="adjacency_matrix", name="Correlation Graph")   
 
 def CompleteBipartiteGraph(p, q, set_position=True):
     r"""

src/sage/graphs/graph_generators.py

@@ -69,6 +69,7 @@ def wrap_name(x):
      "CompleteBipartiteGraph",
      "CompleteGraph",
      "CompleteMultipartiteGraph",
+     "CorrelationGraph",
      "DiamondGraph",
      "GemGraph",
      "DartGraph",
@@ -2355,6 +2356,7 @@ def quadrangulations(self, order, minimum_degree=None, minimum_connectivity=None
     CompleteGraph = staticmethod(basic.CompleteGraph)
     CompleteBipartiteGraph = staticmethod(basic.CompleteBipartiteGraph)
     CompleteMultipartiteGraph = staticmethod(basic.CompleteMultipartiteGraph)
+    CorrelationGraph = staticmethod(basic.CorrelationGraph)
     DiamondGraph = staticmethod(basic.DiamondGraph)
     GemGraph = staticmethod(basic.GemGraph)
     DartGraph = staticmethod(basic.DartGraph)