sty : simplify the reordering of the correlation matrix

nipreps · Mar 24, 2023 · d273c1e · d273c1e
1 parent 00ea0e8
commit d273c1e
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/mriqc_learn/viz/metrics.py b/mriqc_learn/viz/metrics.py
@@ -228,14 +228,8 @@ def plot_corrmat(
         # Keep the indices to sort labels
         labels_order = np.argsort(labels)
 
-        # Build a new dataframe with the sorted columns
-        for idx, i in enumerate(data.columns[labels_order]):
-            if idx == 0:
-                clustered = pd.DataFrame(data[i])
-            else:
-                df_to_append = pd.DataFrame(data[i])
-                clustered = pd.concat([clustered, df_to_append], axis=1)
-        data = clustered
+        # Reorder data
+        data = data.take(labels_order, axis=0).take(labels_order, axis=1)
 
     if hasattr(data, "columns"):
         col_labels = data.columns