Explicitly set dtype of np.lexsort in group_rank (#19679)

pandas-dev · Feb 13, 2018 · 49812cf · 49812cf
1 parent 8cace88
commit 49812cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pandas/_libs/groupby_helper.pxi.in b/pandas/_libs/groupby_helper.pxi.in
@@ -531,7 +531,7 @@ def group_rank_{{name}}(ndarray[float64_t, ndim=2] out,
     # each label corresponds to a different group value,
     # the mask helps you differentiate missing values before
     # performing sort on the actual values
-    _as = np.lexsort(order)
+    _as = np.lexsort(order).view(dtype=np.int64)
 
     if not ascending:
         _as = _as[::-1]