Merge pull request scipy#8177 from andyfaff/gh8152

MAINT: diffev small speedup in ensure constraint
jbn · Nov 20, 2017 · 018ff7a · 018ff7a
2 parents 73f723a + c8d9cdc
commit 018ff7a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/scipy/optimize/_differentialevolution.py b/scipy/optimize/_differentialevolution.py
@@ -689,9 +689,8 @@ def _ensure_constraint(self, trial):
         """
         make sure the parameters lie between the limits
         """
-        for index, param in enumerate(trial):
-            if param > 1 or param < 0:
-                trial[index] = self.random_number_generator.rand()
+        for index in np.where((trial < 0) | (trial > 1))[0]:
+            trial[index] = self.random_number_generator.rand()
 
     def _mutate(self, candidate):
         """