Merge pull request #779 from effigies/fix/restore_confound_order

FIX: Restore confound order in correlation plot
nipreps · Jan 28, 2023 · 283cb36 · 283cb36
2 parents 57caa39 + ad45113
commit 283cb36
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/niworkflows/viz/plots.py b/niworkflows/viz/plots.py
@@ -900,9 +900,9 @@ def confounds_correlation_plot(
     confounds_data = pd.read_table(confounds_file)
 
     if columns:
-        columns = set(columns)  # Drop duplicates
-        columns.add(reference)  # Make sure the reference is included
-        confounds_data = confounds_data[[el for el in columns]]
+        columns = dict.fromkeys(columns)  # Drop duplicates
+        columns[reference] = None  # Make sure the reference is included
+        confounds_data = confounds_data[list(columns)]
 
     confounds_data = confounds_data.loc[
         :, np.logical_not(np.isclose(confounds_data.var(skipna=True), 0))