U
|
O ±dî ã @ s d dl mZ ddlmZ d dlmZ d dlZG dd deZG dd d eZ G d
|
d d eZ
|
d dd ddZ d d d ddZ d d d ddZ d d d ddZdS )é )Ú annotationsé )ÚArray)Ú
|
NamedTupleNc @ s. e Zd ZU ded< ded< ded< ded< dS )ÚUniqueAllResultr ÚvaluesÚindicesÚinverse_indicesÚcountsN©Ú__name__Ú
|
__module__Ú __qualname__Ú__annotations__© r r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/array_api/_set_functions.pyr s
|
r c @ s e Zd ZU ded< ded< dS )ÚUniqueCountsResultr r r
|
Nr r r r r r s
|
r c @ s e Zd ZU ded< ded< dS )ÚUniqueInverseResultr r r Nr r r r r r ! s
|
r r )ÚxÚreturnc C sP t j| jddddd\}}}}| | j¡}tt |¡t |¡t |¡t |¡S )ú|
|
Array API compatible wrapper for :py:func:`np.unique <numpy.unique>`.
|
|
See its docstring for more information.
|
TF©Z return_countsZ return_indexZreturn_inverseZ equal_nan)ÚnpÚuniqueÚ_arrayÚreshapeÚshaper r Ú_new)r r r r r
|
r r r Ú
|
unique_all& s û ür c C s( t j| jddddd}tdd |D S )NTFr c S s g | ]}t |¡qS r )r r )Ú.0Úir r r Ú
|
<listcomp>G s z!unique_counts.<locals>.<listcomp>)r r r r ©r Úresr r r Ú unique_counts> s ûr$ c C s< t j| jddddd\}}| | j¡}tt |¡t |¡S )r FTr )r r r r r r r r )r r r r r r Úunique_inverseJ s û
|
r% c C s t j| jddddd}t |¡S )r Fr )r r r r r r" r r r Ú unique_values] s ûr& )Ú
|
__future__r Z _array_objectr Útypingr Únumpyr r r r r r$ r% r& r r r r Ú<module> s
|
