3
���h � @ s. d dl Zd dlmZ dd� Zed�dd�ZdS )� N)� is_list_likec s� d}t | �st|��x | D ]}t |�st|��qW t| �dkrBg S tjdd� | D �tjd�}tj|�}tj|d�� d� d< |d dkr�|d
| �n
tj|��� �fdd�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 �:/tmp/pip-build-5_djhm0z/pandas/pandas/core/reshape/util.py� <genexpr>' s z$cartesian_product.<locals>.<genexpr>)Zdtype� c s0 g | ](\}}t tj|�| �tj� | ���qS r )� _tile_compat�np�repeat�product)r �ir )�a�br r �
<listcomp>3 s z%cartesian_product.<locals>.<listcomp>���r )
r � TypeErrorr r ZfromiterZintpZ
cumproductZrollZ
zeros_like� enumerate)�X�msgr ZlenXZcumprodXr )r r r �cartesian_product s
r )�numc C s8 t | tj�rtj| |�S tjtjt| ��|�}| j|�S )zf
Index compat for np.tile.
Notes
-----
Does not support multi-dimensional `num`.
)�
isinstancer ZndarrayZtileZaranger Ztake)Zarrr Ztakerr r r r
6 s r
)Znumpyr Zpandas.core.dtypes.commonr r �intr
r r r r �<module> s 0 |