�
���h
� �h � d dl Zd dlZd dlZddlmZmZ d� Zd� Z d� Z
d� Z d
d�Z d
d �Z
d
� Zd � Zd � ZdS )� N� )�builtins�reducec �| � | j }| |cxk r|k sn t d|�d|�d|�d�� � �|dk r||z
}|S )Nzaxis z out of bounds [-z, �)r )�ndim�
IndexError)�data�axisr s �H/opt/cloudlinux/venv/lib/python3.11/site-packages/xarray/core/nputils.py�_validate_axisr
sm � � �9�D� �5�D� � � � �4� � � � ��j� �D�D�$�$�$����.� /� /� /� �a�x�x� �� �� �K� c �~ � t j d� |j D � � � }|d |� |fz ||d � z }| | S )Nc �6 � g | ]}t j |� � ��S � )�np�arange)�.0�ss r �
<listcomp>z&_select_along_axis.<locals>.<listcomp> s � �9�9�9�!���1� � �9�9�9r )r �ix_�shape)�values�idxr � other_ind�sls r �_select_along_axisr sL � ���9�9�s�y�9�9�9�:�I� �5�D�5� �S�F� "�Y�t�u�u�%5� 5�B� �"�:�r c � � t | |� � }t j t j | � � |�� � }t | ||� � S )N�r )r
r �argmax�pd�isnullr )r r � idx_firsts r �nanfirstr$ sD � � �&�$� '� '�D�� �2�9�V�,�,�,�4�8�8�8�I� �f�i�� 6� 6�6r c �� � t | |� � }t d � � f|z t d d d� � fz }dt j t j | � � | |�� � z
}t
| ||� � S )N���r )r
�slicer r r! r" r )r r �rev�idx_lasts r �nanlastr* st � � �&�$� '� '�D� ��;�;�.�4�
�5��t�R�#8�#8�":�
:�C��B�I�r�y�� 0� 0�� 5� 5�D�A�A�A�A�H� �f�h�� 5� 5�5r c � �� | d j }t j �fd�| D � � � � }|d �� |fz |�dz d � z }|S )Nr c 3 �2 �K � | ]}|j � V � �d S )N)r )r �ar s �r � <genexpr>z%_calc_concat_shape.<locals>.<genexpr>' s) �� � � �8�8�A�!�'�$�-�8�8�8�8�8�8r r )r r �sum)�arraysr � first_shape�length� result_shapes ` r �_calc_concat_shaper4 % s_ �� ���)�/�K�
�\�8�8�8�8��8�8�8�
8�
8�F� �u��u�%�� �1�K���� � �4L�L�L� �r c �N � d� | D � � } t | d |� � }t | |�� � }t t j d� | D � � � � }t j ||� � }t
d � � g|j z }t | |� � D ]\ }}|||< |||< �|S )Nc �6 � g | ]}t j |� � ��S r )r �asarray�r r- s r r z&interleaved_concat.<locals>.<listcomp>- s � �
,�
,�
,��b�j��m�m�
,�
,�
,r r r c � � g | ] }|j ��
S r ��dtyper8 s r r z&interleaved_concat.<locals>.<listcomp>0 s � �%>�%>�%>�!�a�g�%>�%>�%>r ) r
r4 r r �
promote_types�emptyr' r �zip) r0 �indicesr r3 r; �result�keyr- �inds r �interleaved_concatrC , s� � �
,�
,�V�
,�
,�
,�F� �&��)�T� *� *�D�%�f�4�8�8�8�L� �2�#�%>�%>�v�%>�%>�%>� ?� ?�E�
�X�l�E�
*�
*�F� ��;�;�-�&�+�
%�C��f�g�&�&� � ���3���D� ���s� � � �Mr c � � t | t � � r?t j |� j }| r t j n t j } ||t �� � } | S )Nr: )�
isinstance�boolr � broadcastr �ones�zeros)r@ �argsr � constructors r �_ensure_bool_is_ndarrayrL 9 sP � � �&�$� � � 0�� �d�#�)��!'�5�b�g�g�R�X� � ��U�$�/�/�/�� �Mr c � � t j � � 5 t j dd� � t | |k | |� � cd d d � � S # 1 swxY w Y d S �N�ignorez elementwise comparison failed��warnings�catch_warnings�filterwarningsrL ��self�others r �array_eqrW G � � � � � "� "� C� C�� ��*J�K�K�K�&�t�u�}�d�E�B�B�C� C� C� C� C� C� C� C� C� C� C� C���� C� C� C� C� C� C� �*A � A�Ac � � t j � � 5 t j dd� � t | |k | |� � cd d d � � S # 1 swxY w Y d S rN rP rT s r �array_ner[ M rX rY )r )�numpyr �pandasr! rQ �pycompatr r r
r r$ r* r4 rC rL rW r[ r r r �<module>r_ s� �� � � � � � � � � ���� &� &� &� &� &� &� &� &�� � �� � � 7� 7� 7� 6� 6� 6�� � � �
�
�
�
� � � � C� C� C� C� C� C� C� Cr |