U ÞYf ã@s$dZddlZd dd„Zd dd„ZdS) z Medoid. éNéFcCsÚ|dkrf|jdd…ddd…f|j}t d||¡}t tjt |¡dd¡}|rV|S|dd…|fS|dkrÈ|dd…ddd…f|}t d||¡}t tjt |¡dd¡}|r¸|S||dd…fStd |¡ƒ‚dS)a Compute the medoid along the specified axis. Returns the medoid of the array elements. Parameters ---------- a : array_like Input array or object that can be converted to an array. axis : int Axis along which the medoid is computed. The default is to compute the median along the last axis of the array. indexonly : bool, optional If this is set to True, only the index of the medoid is returned. Returns ------- medoid : ndarray or int rNú ijk,ijk->ij©Úaxisrúaxis {} out of bounds)ÚTÚnpÚeinsumZargminÚsumÚsqrtÚ IndexErrorÚformat)ÚarÚ indexonlyÚdiffÚssumÚidx©rú/lib/python3.8/site-packages/hdmedians/medoid.pyÚmedoid srcCs|dkr†|jdd…ddd…f|j}t d||¡}tjt |¡dd}t |¡jdd}tj||<t |¡}|rv|S|dd…|fS|dkr |dd…ddd…f|}t d||¡}tjt |¡dd}t |¡jdd}tj||<t |¡}|rú|S||dd…fSt d  |¡ƒ‚dS)a/ Compute the medoid along the specified axis, omitting observations containing NaNs. Returns the medoid of the array elements. Parameters ---------- a : array_like Input array or object that can be converted to an array. axis : int Axis along which the medoid is computed. The default is to compute the median along the last axis of the array. indexonly : bool, optional If this is set to True, only the index of the medoid is returned. Returns ------- medoid : ndarray or int rNrrrr) rrr Znansumr ZisnanÚanyÚnanZ nanargminr r )rrrrrZdistÚmaskrrrrÚ nanmedoid1s*     r)rF)rF)Ú__doc__ZnumpyrrrrrrrÚs (