← Back
Editing: _set_functions.cpython-311.pyc
� ���c� � � � d dl mZ ddlmZ d dlmZ d dlZ G d� de� � Z G d� d e� � Z G d � d e� � Z dd�Z dd�Z dd�Z dd�ZdS )� )� annotations� )�Array)� NamedTupleNc �8 � e Zd ZU ded< ded< ded< ded< dS )�UniqueAllResultr �values�indices�inverse_indices�countsN��__name__� __module__� __qualname__�__annotations__� � �@/usr/lib/python3/dist-packages/numpy/array_api/_set_functions.pyr r s7 � � � � � � ��M�M�M��N�N�N������M�M�M�M�Mr r c �$ � e Zd ZU ded< ded< dS )�UniqueCountsResultr r r Nr r r r r r s" � � � � � � ��M�M�M��M�M�M�M�Mr r c �$ � e Zd ZU ded< ded< dS )�UniqueInverseResultr r r Nr r r r r r ! s' � � � � � � ��M�M�M������r r �xr �returnc �0 � t j | j dddd�� � \ }}}}|� | j � � }t t j |� � t j |� � t j |� � t j |� � � � S )�| Array API compatible wrapper for :py:func:`np.unique <numpy.unique>`. See its docstring for more information. TF�� return_counts� return_index�return_inverse� equal_nan)�np�unique�_array�reshape�shaper r �_new)r r r r r s r � unique_allr( & s� � � 02�y� ������ 0� 0� 0�,�F�G�_�f� &�-�-�a�g�6�6�O� � � �6��� � �7��� � �?�#�#� � �6��� � � r c �d � t j | j dddd�� � }t d� |D � � � S )NTFr c �6 � g | ]}t j |� � ��S r )r r' )�.0�is r � <listcomp>z!unique_counts.<locals>.<listcomp>G s � � ;� ;� ;�!�� �1� � � ;� ;� ;r )r" r# r$ r �r �ress r � unique_countsr0 > sE � � �)� ���� �� � � �C� � ;� ;�s� ;� ;� ;� <�<r c �� � t j | j dddd�� � \ }}|� | j � � }t t j |� � t j |� � � � S )r FTr )r" r# r$ r% r& r r r' )r r r s r �unique_inverser2 J sk � � !�i� ������ � � ��F�O� &�-�-�a�g�6�6�O� �u�z�&� 1� 1�5�:�o�3N�3N� O� O�Or c �f � t j | j dddd�� � }t j |� � S )r Fr )r" r# r$ r r' r. s r � unique_valuesr4 ] s; � � �)� ���� �� � � �C� �:�c�?�?�r )r r r r )r r r r )r r r r )r r r r )� __future__r � _array_objectr �typingr �numpyr" r r r r( r0 r2 r4 r r r �<module>r9 s# �� "� "� "� "� "� "� � � � � � � � � � � � � � � � � � � � � �j� � � �� � � � �� � � � � � � � �*� � � � � � � �0 =� =� =� =�P� P� P� P�& � � � � � r
Save File
Cancel