make get_heatmap less prone to mess up transposing

BojarLab · Jun 12, 2024 · c68fcd2 · c68fcd2
1 parent 754d543
commit c68fcd2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/build/lib/glycowork/motif/analysis.py b/build/lib/glycowork/motif/analysis.py
@@ -238,7 +238,7 @@ def get_heatmap(df, motifs = False, feature_set = ['known'], transform = '',
     df = df.set_index(index_col)
   elif isinstance(df.iloc[0,0], str):
     df = df.set_index(df.columns.tolist()[0])
-  if not isinstance(df.index.tolist()[0], str) or (isinstance(df.index.tolist()[0], str) and '(' not in df.index.tolist()[0] and '-' not in df.index.tolist()[0]):
+  if not isinstance(df.index[0], str) or (isinstance(df.index[0], str) and ('(' not in df.index[0] or '-' not in df.index[0])):
     df = df.T
   df = df.fillna(0)
   if transform == "CLR":

diff --git a/glycowork/motif/analysis.py b/glycowork/motif/analysis.py
@@ -238,7 +238,7 @@ def get_heatmap(df, motifs = False, feature_set = ['known'], transform = '',
     df = df.set_index(index_col)
   elif isinstance(df.iloc[0,0], str):
     df = df.set_index(df.columns.tolist()[0])
-  if not isinstance(df.index.tolist()[0], str) or (isinstance(df.index.tolist()[0], str) and '(' not in df.index.tolist()[0] and '-' not in df.index.tolist()[0]):
+  if not isinstance(df.index[0], str) or (isinstance(df.index[0], str) and ('(' not in df.index[0] or '-' not in df.index[0])):
     df = df.T
   df = df.fillna(0)
   if transform == "CLR":