add continuous value for module_trait_relationships_heatmap()

mortazavilab · Feb 1, 2024 · 3c7694d · 3c7694d
1 parent 1feba73
commit 3c7694d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/PyWGCNA/wgcna.py b/PyWGCNA/wgcna.py
@@ -25,6 +25,7 @@
 from gseapy.plot import dotplot
 from pyvis.network import Network
 from reactome2py import analysis
+from pandas.api.types import is_numeric_dtype
 
 from PyWGCNA.geneExp import *
 
@@ -2734,6 +2735,9 @@ def getDatTraits(self, metaData):
         data = self.datExpr.obs.copy()[metaData]
         datTraits = pd.DataFrame(index=data.index)
         for i in range(data.shape[1]):
+            if is_numeric_dtype(data.iloc[:, i].dtypes):
+                datTraits[data.columns[i]] = data.iloc[:, i]
+                continue
             data.iloc[:, i] = data.iloc[:, i].astype(str)
             if len(np.unique(data.iloc[:, i])) == 2:
                 datTraits[data.columns[i]] = data.iloc[:, i]