B W0d~@s.ddlZddlmZddZedddZdS)N) is_list_likecsd}t|st|x|D]}t|st|qWt|dkrBgStjdd|Dtjd}t|}t|dkr|tdt |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)rir)abrr 6sz%cartesian_product..) r TypeErrorrr ZfromiterZintpZ cumproductany ValueErrorZrollZ zeros_like enumerate)XmsgrZlenXZcumprodXr)rrrcartesian_products$       r)numcCs8t|tjrt||Sttt||}||S)zf Index compat for np.tile. Notes ----- Does not support multi-dimensional `num`. ) isinstancer ZndarrayZtileZarangerZtake)ZarrrZtakerrrrr 9s  r )numpyr Zpandas.core.dtypes.commonrrintr rrrrs 3