STY: Apply ruff/Perflint rule PERF401

PERF401 Use a list comprehension to create a transformed list

Author: DimitriPapadopoulos

nipype/algorithms/confounds.py

@@ -1289,11 +1289,8 @@ def combine_mask_files(mask_files, mask_method=None, mask_index=None):
         raise ValueError(
             f"mask_index {mask_index} must be less than number of mask files {len(mask_files)}"
         )
-    masks = []
     if mask_method == "none":
-        for filename in mask_files:
-            masks.append(nb.load(filename))
-        return masks
+        return [nb.load(filename) for filename in mask_files]
 
     if mask_method == "union":
         mask = None

nipype/algorithms/misc.py

@@ -548,11 +548,9 @@ def maketypelist(rowheadings, shape, extraheadingBool, extraheading):
     if rowheadings:
         typelist.append(("heading", "a40"))
     if len(shape) > 1:
-        for idx in range(1, (min(shape) + 1)):
-            typelist.append((str(idx), float))
+        typelist.extend((str(idx), float) for idx in range(1, (min(shape) + 1)))
     else:
-        for idx in range(1, (shape[0] + 1)):
-            typelist.append((str(idx), float))
+        typelist.extend((str(idx), float) for idx in range(1, (shape[0] + 1)))
     if extraheadingBool:
         typelist.append((extraheading, "a40"))
     iflogger.info(typelist)
@@ -713,8 +711,7 @@ def _run_interface(self, runtime):
                 mx = shape[0]
             else:
                 mx = 1
-            for idx in range(0, mx):
-                extrafieldlist.append(self.inputs.extra_field)
+            extrafieldlist.extend(self.inputs.extra_field for idx in range(0, mx))
             iflogger.info(len(extrafieldlist))
             output[extraheading] = extrafieldlist
         iflogger.info(output)

nipype/algorithms/modelgen.py

@@ -473,15 +473,15 @@ def _generate_design(self, infolist=None):
         """Generate design specification for a typical fmri paradigm"""
         realignment_parameters = []
         if isdefined(self.inputs.realignment_parameters):
-            for parfile in self.inputs.realignment_parameters:
-                realignment_parameters.append(
-                    np.apply_along_axis(
-                        func1d=normalize_mc_params,
-                        axis=1,
-                        arr=np.loadtxt(parfile),
-                        source=self.inputs.parameter_source,
-                    )
+            realignment_parameters.extend(
+                np.apply_along_axis(
+                    func1d=normalize_mc_params,
+                    axis=1,
+                    arr=np.loadtxt(parfile),
+                    source=self.inputs.parameter_source,
                 )
+                for parfile in self.inputs.realignment_parameters
+            )
         outliers = []
         if isdefined(self.inputs.outlier_files):
             for filename in self.inputs.outlier_files:

nipype/interfaces/ants/registration.py

@@ -190,10 +190,11 @@ def _transformation_constructor(self):
         delta_time = self.inputs.delta_time
         symmetry_type = self.inputs.symmetry_type
         retval = ["--transformation-model %s" % model]
-        parameters = []
-        for elem in (step_length, time_step, delta_time, symmetry_type):
-            if elem is not traits.Undefined:
-                parameters.append("%#.2g" % elem)
+        parameters = [
+            "%#.2g" % elem
+            for elem in (step_length, time_step, delta_time, symmetry_type)
+            if elem is not traits.Undefined
+        ]
         if len(parameters) > 0:
             if len(parameters) > 1:
                 parameters = ",".join(parameters)
@@ -1136,8 +1137,7 @@ def _format_registration(self):
         retval = []
         for ii in range(len(self.inputs.transforms)):
             retval.append("--transform %s" % (self._format_transform(ii)))
-            for metric in self._format_metric(ii):
-                retval.append("--metric %s" % metric)
+            retval.extend("--metric %s" % metric for metric in self._format_metric(ii))
             retval.append("--convergence %s" % self._format_convergence(ii))
             if isdefined(self.inputs.sigma_units):
                 retval.append(

nipype/interfaces/ants/segmentation.py

@@ -805,16 +805,15 @@ def _list_outputs(self):
             os.getcwd(),
             self.inputs.out_prefix + "BrainSegmentation0N4." + self.inputs.image_suffix,
         )
-        posteriors = []
-        for i in range(len(self.inputs.segmentation_priors)):
-            posteriors.append(
-                os.path.join(
-                    os.getcwd(),
-                    self.inputs.out_prefix
-                    + "BrainSegmentationPosteriors%02d." % (i + 1)
-                    + self.inputs.image_suffix,
-                )
+        posteriors = [
+            os.path.join(
+                os.getcwd(),
+                self.inputs.out_prefix
+                + "BrainSegmentationPosteriors%02d." % (i + 1)
+                + self.inputs.image_suffix,
             )
+            for i in range(len(self.inputs.segmentation_priors))
+        ]
         outputs["BrainSegmentationPosteriors"] = posteriors
         outputs["CorticalThickness"] = os.path.join(
             os.getcwd(),
@@ -1480,15 +1479,13 @@ class JointFusion(ANTSCommand):
 
     def _format_arg(self, opt, spec, val):
         if opt == "exclusion_image_label":
-            retval = []
-            for ii in range(len(self.inputs.exclusion_image_label)):
-                retval.append(
-                    "-e {}[{}]".format(
-                        self.inputs.exclusion_image_label[ii],
-                        self.inputs.exclusion_image[ii],
-                    )
+            return " ".join(
+                "-e {}[{}]".format(
+                    self.inputs.exclusion_image_label[ii],
+                    self.inputs.exclusion_image[ii],
                 )
-            return " ".join(retval)
+                for ii in range(len(self.inputs.exclusion_image_label))
+            )
         if opt == "patch_radius":
             return f"-p {self._format_xarray(val)}"
         if opt == "search_radius":

nipype/interfaces/base/core.py

@@ -1047,10 +1047,11 @@ def __init__(self, check_import=True, *args, **kwargs):
         if check_import:
             import pkgutil
 
-            failed_imports = []
-            for pkg in (self._pkg,) + tuple(self.imports):
-                if pkgutil.find_loader(pkg) is None:
-                    failed_imports.append(pkg)
+            failed_imports = [
+                pkg
+                for pkg in (self._pkg,) + tuple(self.imports)
+                if pkgutil.find_loader(pkg) is None
+            ]
             if failed_imports:
                 iflogger.warning(
                     "Unable to import %s; %s interface may fail to run",

nipype/interfaces/cmtk/cmtk.py

@@ -328,9 +328,11 @@ def cmat(
     else:
         final_fibers_indices = final_fibers_idx
 
-    for idx in final_fibers_indices:
+    finalfiberlength.extend(
         # compute length of fiber
-        finalfiberlength.append(length(fib[idx][0]))
+        length(fib[idx][0])
+        for idx in final_fibers_indices
+    )
 
     # convert to array
     final_fiberlength_array = np.array(finalfiberlength)
@@ -463,9 +465,7 @@ def cmat(
 def save_fibers(oldhdr, oldfib, fname, indices):
     """Stores a new trackvis file fname using only given indices"""
     hdrnew = oldhdr.copy()
-    outstreams = []
-    for i in indices:
-        outstreams.append(oldfib[i])
+    outstreams = [oldfib[i] for i in indices]
     n_fib_out = len(outstreams)
     hdrnew["n_count"] = n_fib_out
     iflogger.info("Writing final non-orphan fibers as %s", fname)

nipype/interfaces/dipy/simulate.py

@@ -221,20 +221,17 @@ def _run_interface(self, runtime):
         mevals = [sf_evals] * nsticks + [[ba_evals[d]] * 3 for d in range(nballs)]
 
         b0 = b0_im.get_fdata()[msk > 0]
-        args = []
-        for i in range(nvox):
-            args.append(
-                {
-                    "fractions": fracs[i, ...].tolist(),
-                    "sticks": [
-                        tuple(dirs[i, j : j + 3]) for j in range(nsticks + nballs)
-                    ],
-                    "gradients": gtab,
-                    "mevals": mevals,
-                    "S0": b0[i],
-                    "snr": self.inputs.snr,
-                }
-            )
+        args = [
+            {
+                "fractions": fracs[i, ...].tolist(),
+                "sticks": [tuple(dirs[i, j : j + 3]) for j in range(nsticks + nballs)],
+                "gradients": gtab,
+                "mevals": mevals,
+                "S0": b0[i],
+                "snr": self.inputs.snr,
+            }
+            for i in range(nvox)
+        ]
 
         n_proc = self.inputs.n_proc
         if n_proc == 0:

nipype/interfaces/freesurfer/preprocess.py

@@ -593,10 +593,10 @@ def _list_outputs(self):
                     raise Exception(
                         "Not taking frame manipulations into account- please warn the developers"
                     )
-                outfiles = []
                 outfile = self._get_outfilename()
-                for i in range(tp):
-                    outfiles.append(fname_presuffix(outfile, suffix="%03d" % (i + 1)))
+                outfiles = [
+                    fname_presuffix(outfile, suffix="%03d" % (i + 1)) for i in range(tp)
+                ]
                 outfile = outfiles
         outputs["out_file"] = outfile
         return outputs

nipype/interfaces/fsl/fix.py

@@ -105,11 +105,11 @@ class TrainingSetCreator(BaseInterface):
     _always_run = True
 
     def _run_interface(self, runtime):
-        mel_icas = []
-        for item in self.inputs.mel_icas_in:
-            if os.path.exists(os.path.join(item, "hand_labels_noise.txt")):
-                mel_icas.append(item)
-
+        mel_icas = [
+            item
+            for item in self.inputs.mel_icas_in
+            if os.path.exists(os.path.join(item, "hand_labels_noise.txt"))
+        ]
         if len(mel_icas) == 0:
             raise Exception(
                 "%s did not find any hand_labels_noise.txt files in the following directories: %s"
@@ -119,10 +119,11 @@ def _run_interface(self, runtime):
         return runtime
 
     def _list_outputs(self):
-        mel_icas = []
-        for item in self.inputs.mel_icas_in:
-            if os.path.exists(os.path.join(item, "hand_labels_noise.txt")):
-                mel_icas.append(item)
+        mel_icas = [
+            item
+            for item in self.inputs.mel_icas_in
+            if os.path.exists(os.path.join(item, "hand_labels_noise.txt"))
+        ]
         outputs = self._outputs().get()
         outputs["mel_icas_out"] = mel_icas
         return outputs