BUG: group_shift_indexer checks for null group keys

pandas-dev · Jul 29, 2016 · 966d5c6 · 966d5c6
1 parent 2c55f28
commit 966d5c6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pandas/src/algos_groupby_helper.pxi b/pandas/src/algos_groupby_helper.pxi
@@ -1356,6 +1356,11 @@ def group_shift_indexer(int64_t[:] out, int64_t[:] labels,
                 ## reverse iterator if shifting backwards
                 ii = offset + sign * i
                 lab = labels[ii]
+
+                # Skip null keys
+                if lab == -1:
+                    continue
+
                 label_seen[lab] += 1
 
                 idxer_slot = label_seen[lab] % periods