update clusterprofile utility

Author: jefferyUstc

dotplot/utils.py

@@ -1,8 +1,12 @@
 import numpy as np
 import pandas as pd
 
+DEFAULT_CLUSTERPROFILE_KEYS = {
+    'item_key': 'Description', 'group_key': 'group', 'sizes_key': 'GeneRatio', 'color_key': 'pvalue'
+}
 
-def merge_clusterprofile_results(dataframes, groups, term_list=None):
+
+def merge_clusterprofile_results(dataframes, groups, group_key='group', term_list=None):
     assert len(dataframes) == len(groups)
     merged_df = None
     for _dataframe, _group in zip(dataframes, groups):
@@ -11,12 +15,12 @@ def merge_clusterprofile_results(dataframes, groups, term_list=None):
         else:
             _sub_df = _dataframe[_dataframe.index.isin(term_list)]
         if not _sub_df.empty:
-            _sub_df['group'] = _group
+            _sub_df[group_key] = _group
             if merged_df is not None:
                 merged_df = pd.concat((merged_df, _sub_df))
             else:
                 merged_df = _sub_df
-    merged_df = merged_df[['Description', 'pvalue', 'GeneRatio', 'group']]
+    merged_df = merged_df[['Description', 'pvalue', 'GeneRatio', group_key]]
     merged_df['GeneRatio'] = merged_df.GeneRatio.map(lambda x: int(x.split('/')[0]))
     merged_df['pvalue'] = merged_df['pvalue'].map(lambda x: -np.log10(x))
     return merged_df