3
���h
� @ sx d dl Zd dlZd dlZddlmZmZ dd� Zdd� Z dd � Z
d
d � Z dd d
�Z ddd�Z
dd� Zdd� Zdd� ZdS )� N� )�builtins�reducec C sF | j }| | ko|k n s2td|||f ��|dk rB||7 }|S )Nz axis %r out of bounds [-%r, %r)r )�ndim�
IndexError)�data�axisr � r �3/tmp/pip-build-5_djhm0z/xray/xarray/core/nputils.py�_validate_axis s r c C s< t jdd� |jD �� }|d |� |f ||d � }| | S )Nc S s g | ]}t j|��qS r )�npZarange)�.0�sr r r
�
<listcomp> s z&_select_along_axis.<locals>.<listcomp>)r Zix_�shape)�values�idxr Z other_ind�slr r r
�_select_along_axis s r c C s, t | |�}tjtj| � |d�}t| ||�S )N)r )r r �argmax�pd�isnullr )r r Z idx_firstr r r
�nanfirst s
r c C sP t | |�}td �f| td d d�f }dtjtj| �| |d� }t| ||�S )Nr )r ���r )r �slicer r r r r )r r �revZidx_lastr r r
�nanlast s
r c sH | d j }tj� fdd�| D ��}|d � � |f |� d d � }|S )Nr c 3 s | ]}|j � V qd S )N)r )r
�a)r r r
� <genexpr>' s z%_calc_concat_shape.<locals>.<genexpr>r )r r �sum)�arraysr Z first_shape�length� result_shaper )r r
�_calc_concat_shape% s
"r# c C s� dd� | D �} t | d |�}t| |d�}ttjdd� | D ��}tj||�}td �g|j }x&t| |�D ]\}}|||< |||< qfW |S )Nc S s g | ]}t j|��qS r )r Zasarray)r
r r r r
r - s z&interleaved_concat.<locals>.<listcomp>r )r c S s g | ]
}|j �qS r )�dtype)r
r r r r
r 0 s ) r r# r r Z
promote_types�emptyr r �zip) r �indicesr r" r$ �result�keyr �indr r r
�interleaved_concat, s r+ c G s6 t | t�r2tj|� j}| r tjntj}||td�} | S )N)r$ )�
isinstance�boolr � broadcastr Zones�zeros)r( �argsr � constructorr r r
�_ensure_bool_is_ndarray9 s
r2 c C s0 t j� � t jdd� t| |k| |�S Q R X d S )N�ignorez elementwise comparison failed)�warnings�catch_warnings�filterwarningsr2 )�self�otherr r r
�array_eqG s
r9 c C s0 t j� � t jdd� t| |k| |�S Q R X d S )Nr3 z elementwise comparison failed)r4 r5 r6 r2 )r7 r8 r r r
�array_neM s
r: )r )r )Znumpyr Zpandasr r4 Zpycompatr r r r r r r# r+ r2 r9 r: r r r r
�<module> s
|