Skip to content

Commit f7e910e

Browse files
Gabe SmallGabe Small
authored andcommitted
shortened factorize
1 parent cc1deb6 commit f7e910e

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

pandas/core/reshape/reshape.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -175,12 +175,8 @@ def _indexer_and_to_sort(
175175
if not self.sort:
176176
# Create new codes considering that labels are already sorted
177177
# Make sure to preserve the -1 values before factorizing
178-
codes = []
179-
for code in self.index.codes:
180-
mask = code != -1
181-
factorized = np.full_like(code, -1)
182-
factorized[mask] = factorize(code[mask])[0]
183-
codes.append(factorized)
178+
codes = [factorize(np.where(code == -1, None, code))[0] for code in codes]
179+
184180
levs = list(self.index.levels)
185181
to_sort = codes[:v] + codes[v + 1 :] + [codes[v]]
186182
sizes = tuple(len(x) for x in levs[:v] + levs[v + 1 :] + [levs[v]])

0 commit comments

Comments
 (0)