Skip to content

Commit a0ecaad

Browse files
authored
Fix MultiIndex zip with strict argument
1 parent 4d83483 commit a0ecaad

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pandas/core/reshape/reshape.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -778,7 +778,8 @@ def _stack_multi_column_index(columns: MultiIndex) -> MultiIndex | Index:
778778

779779
levs = (
780780
[lev[c] if c >= 0 else None for c in codes]
781-
for lev, codes in zip(columns.levels[:-1], columns.codes[:-1])
781+
for lev, codes in zip(columns.levels[:-1], columns.codes[:-1], strict-True)
782+
782783
)
783784

784785
# Remove duplicate tuples in the MultiIndex.
@@ -792,7 +793,7 @@ def _stack_multi_column_index(columns: MultiIndex) -> MultiIndex | Index:
792793
[
793794
# Not all indices can accept None values.
794795
Index(new_lev, dtype=lev.dtype) if None not in new_lev else new_lev
795-
for new_lev, lev in zip(new_levs, columns.levels)
796+
for new_lev, lev in zip(new_levs, columns.levels[:-1], strict=True)
796797
],
797798
names=columns.names[:-1],
798799
)

0 commit comments

Comments
 (0)