@@ -183,7 +183,9 @@ def check_len(item, name: str) -> None:
183183 check_len (prefix_sep , "prefix_sep" )
184184
185185 if isinstance (prefix , str ):
186- prefix = itertools .cycle ([prefix ])
186+ prefix = itertools .islice (
187+ itertools .cycle ([prefix ]), data_to_encode .shape [1 ]
188+ )
187189 if isinstance (prefix , dict ):
188190 prefix = [prefix [col ] for col in data_to_encode .columns ]
189191
@@ -192,7 +194,9 @@ def check_len(item, name: str) -> None:
192194
193195 # validate separators
194196 if isinstance (prefix_sep , str ):
195- prefix_sep = itertools .cycle ([prefix_sep ])
197+ prefix_sep = itertools .islice (
198+ itertools .cycle ([prefix_sep ]), data_to_encode .shape [1 ]
199+ )
196200 elif isinstance (prefix_sep , dict ):
197201 prefix_sep = [prefix_sep [col ] for col in data_to_encode .columns ]
198202
@@ -210,7 +214,7 @@ def check_len(item, name: str) -> None:
210214 with_dummies = [data .select_dtypes (exclude = dtypes_to_encode )]
211215
212216 for col , pre , sep in zip (
213- data_to_encode .items (), prefix , prefix_sep , strict = False
217+ data_to_encode .items (), prefix , prefix_sep , strict = True
214218 ):
215219 # col is (column_name, column), use just column data here
216220 dummy = _get_dummies_1d (
0 commit comments