add a plot function

FunGI-cap · Oct 8, 2023 · a071f8d · a071f8d
1 parent 4627dee
commit a071f8d
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/gimaps/heatmap.py b/gimaps/heatmap.py
@@ -0,0 +1,22 @@
+import PyComplexHeatmap as ch
+
+
+def plot_heatmap(data,ax,leg=True,vmin=-.4, vmax=.4):
+    plot_cm = ch.ClusterMapPlotter(
+        data,
+        col_dendrogram=True,
+        row_dendrogram=True,
+        # left_annotation=col_ha,
+        vmin=vmin, vmax=vmax,
+        # center = 0,
+        # rasterized=True, 
+        cmap = 'BlueYellow',
+        verbose=False,
+        legend=leg,legend_gap=1,
+        plot=False
+        # show_rownames=True,show_colnames=True
+    )
+    plot_cm.plot(ax=ax)
+    plot_cm.plot_legends(ax=ax)
+
+    return ax