added binary_motif_mtx fn

colomemaria · Jan 29, 2024 · 939f93e · 939f93e
1 parent a940ea5
commit 939f93e
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/episcanpy/tools/_motif_search.py b/episcanpy/tools/_motif_search.py
@@ -183,3 +183,13 @@ def run(self) -> None:
         self.q.put(results)
 
         self.sem.release()
+
+
+def binary_motif_mtx(adata):
+
+    motif_presence_mtx = np.zeros((len(adata.uns["motif_search"]["tf_motifs"]), adata.n_vars))
+
+    for tfbs in adata.uns["motif_search"]["results"]:
+        motif_presence_mtx[tfbs["motif_idx"], tfbs["feature_idx"]] = 1
+
+    adata.uns["motif_search"]["motif_presence_mtx"] = motif_presence_mtx