B
    2c                 @  sH   d dl mZ d dlmZmZ dddZdddZdd	d
ddddZdS )    )annotations)	DataFrameconcatNc               s   ddddd}t  tr0 fdd| D }nt  trD|  |drd| jd |d ksdt|dr| jd |d kst|d	r| jd	 |d	 kst|d
r| jd
 |d
 kstd S )NZfloat32Zfloat16Zfloat64)ABCDc               s   i | ]\}} |qS  r	   ).0kv)dtyper	   =/tmp/pip-install-v41xt_bg/pandas/pandas/tests/frame/common.py
<dictcomp>   s    z&_check_mixed_float.<locals>.<dictcomp>r   r   r   r   )
isinstancestritemsdictupdategetdtypesAssertionError)dfr   r   r	   )r   r   _check_mixed_float	   s    






r   c               s   ddddd}t  tr0 fdd| D }nt  trD|  |drd| jd |d ksdt|d	r| jd	 |d	 kst|d
r| jd
 |d
 kst|dr| jd |d kstd S )NZint32Zuint64Zuint8Zint64)r   r   r   r   c               s   i | ]\}} |qS r	   r	   )r
   r   r   )r   r	   r   r      s    z$_check_mixed_int.<locals>.<dictcomp>r   r   r   r   )r   r   r   r   r   r   r   r   )r   r   r   r	   )r   r   _check_mixed_int   s    






r      zlist[DataFrame]intr   )framesaxisreturnc               sX   |dkr0 d j } fdd|D }t|ddS  d j} fdd|D }t|S dS )z
    take a list of frames, zip them together under the
    assumption that these all have the first frames' index/columns.

    Returns
    -------
    new_frame : DataFrame
    r   r   c               s(   g | ] } D ]}|j d d |f qqS )N)loc)r
   cf)r   r	   r   
<listcomp>5   s    zzip_frames.<locals>.<listcomp>)r   c               s(   g | ] } D ]}|j |d d f qqS )N)r    )r
   ir"   )r   r	   r   r#   9   s    N)columnsr   indexr   )r   r   r%   Zzippedr&   r	   )r   r   
zip_frames*   s    	

r'   )N)N)r   )
__future__r   Zpandasr   r   r   r   r'   r	   r	   r	   r   <module>   s   

