FIX for len(result) > minlength

amueller · amueller · commit d6655fa8e0b1 · 2012-10-12T09:16:54.000+01:00
diff --git a/sklearn/utils/fixes.py b/sklearn/utils/fixes.py
@@ -109,8 +109,10 @@ def _unique(ar, return_index=False, return_inverse=False):
 
 def _bincount(X, weights=None, minlength=None):
     """Replacing np.bincount in numpy < 1.6 to provide minlength."""
-    out = np.zeros(minlength, np.int)
     result = np.bincount(X, weights)
+    if len(result) >= minlength:
+        return result
+    out = np.zeros(minlength, np.int)
     out[:len(result)] = result
     return out
 
