a Of@s`ddlmZddlmZddlZddlmZergStjdd|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='.z$cartesian_product..)Zdtypez+Product space too large to allocate arrays!cs0g|](\}}tt||t|qSr ) tile_compatnprepeatprod)rir abr r ?s  z%cartesian_product..) r TypeErrorrrZfromiterZintpZcumprodany ValueErrorZrollZ zeros_like enumerate)Xmsgr ZlenXZcumprodXr rr cartesian_product s(       rrint)arrnumrcCs8t|tjrt||Sttt||}||S)zf Index compat for np.tile. Notes ----- Does not support multi-dimensional `num`. ) isinstancerZndarrayZtileZarangerZtake)r!r"Ztakerr r r rHs  r) __future__rtypingrnumpyrZpandas.core.dtypes.commonrZpandas._typingrrrr r r r s    ;