update

tmichoel · Feb 16, 2024 · 6489c00 · 6489c00
1 parent 336d83b
commit 6489c00
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/bayesiannets.jl b/src/bayesiannets.jl
@@ -0,0 +1,11 @@
+"""
+    splitAB(dP)
+
+Split a DataFrame of [`findr`](@ref) results (list of edges) into two DataFrames, one with all `Source` to `Source` edges, and the other with all `Source` to `non-Source` edges. 
+"""
+function splitAB(dP)
+    source_names = unique(dP.Source);
+    dP1 = filter(row -> row.Target ∈ source_names, dP)
+    dP2 = filter(row -> row.Target ∉ source_names, dP) 
+    return dP1, dP2
+end