U
|
¬ý°d$ ã @ sP d dl mZ d dlZd dlmZ d dlmZ ddddZd d
|
d d d d Z dS )é )Ú annotationsN)Ú NumpyIndexT)Ú is_list_likezlist[np.ndarray])Úreturnc sÈ d}t | st|| D ]}t |st|qt| dkr>g S tjdd | D tjd}t |¡}t |dk ¡rxtdt |d¡ d d< |d dkr¦|d | n
|
t
|
|¡ fd d
|
t | D S ) 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='<U1'), array([1, 2, 1, 2, 1, 2])]
|
|
See Also
|
--------
|
itertools.product : Cartesian product of input iterables. Equivalent to
|
nested for-loops.
|
z'Input must be a list-like of list-likesr c s s | ]}t |V qd S )N)Úlen)Ú.0Úx© r úOd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/reshape/util.pyÚ <genexpr>+ s z$cartesian_product.<locals>.<genexpr>)Zdtypez+Product space too large to allocate arrays!é éÿÿÿÿc s0 g | ](\}}t t || ¡t | ¡qS r )Ú tile_compatÚnpÚrepeatÚprod)r Úir ©ÚaÚbr r
|
Ú
|
<listcomp>< s
|
ü þz%cartesian_product.<locals>.<listcomp>) r Ú TypeErrorr r ZfromiterZintpZcumprodÚanyÚ
|
ValueErrorZrollZ
|
zeros_likeÚ enumerate)ÚXÚmsgr ZlenXZcumprodXr r r
|
Úcartesian_product
|
s(
|
|
|
ûr r Úint)ÚarrÚnumr c C s8 t | tjrt | |¡S t t t| ¡|¡}| |¡S )zf
|
Index compat for np.tile.
|
|
Notes
|
-----
|
Does not support multi-dimensional `num`.
|
)Ú
|
isinstancer ZndarrayZtileZaranger Ztake)r r Ztakerr r r
|
r E s r )
|
Ú
|
__future__r Únumpyr Zpandas._typingr Zpandas.core.dtypes.commonr r r r r r r
|
Ú<module> s
|
;
|
