Skip to content

Commit 382faa5

Browse files
authored
STY: Add strict=True to zip() calls in pandas/core/reshape/concat.py (Ruff B905)
1 parent 607e489 commit 382faa5

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pandas/core/reshape/concat.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -840,7 +840,7 @@ def _make_concat_multiindex(indexes, keys, levels=None, names=None) -> MultiInde
840840
if (levels is None and isinstance(keys[0], tuple)) or (
841841
levels is not None and len(levels) > 1
842842
):
843-
zipped = list(zip(*keys))
843+
zipped = list(zip(*keys, strict=True))
844844
if names is None:
845845
names = [None] * len(zipped)
846846

@@ -866,13 +866,13 @@ def _make_concat_multiindex(indexes, keys, levels=None, names=None) -> MultiInde
866866
# things are potentially different sizes, so compute the exact codes
867867
# for each level and pass those to MultiIndex.from_arrays
868868

869-
for hlevel, level in zip(zipped, levels):
869+
for hlevel, level in zip(zipped, levels, strict=True):
870870
to_concat = []
871871
if isinstance(hlevel, Index) and hlevel.equals(level):
872872
lens = [len(idx) for idx in indexes]
873873
codes_list.append(np.repeat(np.arange(len(hlevel)), lens))
874874
else:
875-
for key, index in zip(hlevel, indexes):
875+
for key, index in zip(hlevel, indexes, strict=True):
876876
# Find matching codes, include matching nan values as equal.
877877
mask = (isna(level) & isna(key)) | (level == key)
878878
if not mask.any():
@@ -922,7 +922,7 @@ def _make_concat_multiindex(indexes, keys, levels=None, names=None) -> MultiInde
922922

923923
# do something a bit more speedy
924924

925-
for hlevel, level in zip(zipped, levels):
925+
for hlevel, level in zip(zipped, levels, strict=True):
926926
hlevel_index = ensure_index(hlevel)
927927
mapped = level.get_indexer(hlevel_index)
928928

0 commit comments

Comments
 (0)