FIX: Restore confound order in correlation plot

nipreps · Jan 27, 2023 · ad45113 · ad45113
1 parent 57caa39
commit ad45113
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))