55 Hashable ,
66 Iterable ,
77)
8- import itertools
98from typing import TYPE_CHECKING
109
1110import numpy as np
@@ -185,7 +184,7 @@ def check_len(item, name: str) -> None:
185184 check_len (prefix_sep , "prefix_sep" )
186185
187186 if isinstance (prefix , str ):
188- prefix = itertools . cycle ( [prefix ])
187+ prefix = [prefix ] * len ( data_to_encode . columns )
189188 if isinstance (prefix , dict ):
190189 prefix = [prefix [col ] for col in data_to_encode .columns ]
191190
@@ -194,7 +193,7 @@ def check_len(item, name: str) -> None:
194193
195194 # validate separators
196195 if isinstance (prefix_sep , str ):
197- prefix_sep = itertools . cycle ( [prefix_sep ])
196+ prefix_sep = [prefix_sep ] * len ( data_to_encode . columns )
198197 elif isinstance (prefix_sep , dict ):
199198 prefix_sep = [prefix_sep [col ] for col in data_to_encode .columns ]
200199
@@ -212,7 +211,7 @@ def check_len(item, name: str) -> None:
212211 with_dummies = [data .select_dtypes (exclude = dtypes_to_encode )]
213212
214213 for col , pre , sep in zip (
215- data_to_encode .items (), prefix , prefix_sep , strict = False
214+ data_to_encode .items (), prefix , prefix_sep , strict = True
216215 ):
217216 # col is (column_name, column), use just column data here
218217 dummy = _get_dummies_1d (
0 commit comments