U ޙe @s$dZddlZd ddZd ddZdS) z Medoid. NFcCs|dkrf|jdddddf|j}td||}ttjt|dd}|rV|S|dd|fS|dkr|dddddf|}td||}ttjt|dd}|r|S||ddfStd|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->ijaxisraxis {} out of bounds)TnpeinsumZargminsumsqrt IndexErrorformat)ar indexonlydiffssumidxr/lib/python3.8/site-packages/hdmedians/medoid.pymedoid srcCs|dkr|jdddddf|j}td||}tjt|dd}t|jdd}tj||<t|}|rv|S|dd|fS|dkr |dddddf|}td||}tjt|dd}t|jdd}tj||<t|}|r|S||ddfSt 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 ZisnananynanZ nanargminr r )rrrrrZdistmaskrrrr nanmedoid1s*     r)rF)rF)__doc__Znumpyrrrrrrrs (