U f/e~@s.ddlZddlmZddZedddZdS)N) is_list_likecsd}t|st||D]}t|st|qt|dkr>gStjdd|Dtjd}t|}t|dkrxtdt |ddd<|ddkr|d|n t |fd d t |DS) a Numpy version of itertools.product. Sometimes faster (for large inputs)... Parameters ---------- X : list-like of list-likes Returns ------- product : list of ndarrays Examples -------- >>> cartesian_product([list('ABC'), [1, 2]]) [array(['A', 'A', 'B', 'B', 'C', 'C'], dtype=''sz$cartesian_product..)Zdtypez+Product space too large to allocate arrays!cs0g|](\}}tt||t|qSr) tile_compatnprepeatproduct)rirabrr 6sz%cartesian_product..) r TypeErrorrr ZfromiterZintpZ cumproductany ValueErrorZrollZ zeros_like enumerate)XmsgrZlenXZcumprodXrrrcartesian_products$      r)numcCs8t|tjrt||Sttt||}||S)zf Index compat for np.tile. Notes ----- Does not support multi-dimensional `num`. ) isinstancer ZndarrayZtileZarangerZtake)ZarrrZtakerrrrr 9s  r )Znumpyr Zpandas.core.dtypes.commonrrintr rrrrs 3