номера карты в их процентили

Question

Я хотел бы применитьрезультата от numpy.percentile к своему аргументу, то есть карта каждого числа во входном векторе ее квантиль.

Например, если v=np.array([1,2,3,4]), и я хочу только два квантили (больше и меньше, чем медиана), я хотел бы получить np.array([0,0,1,1]) рассказывал мне, что первые два элемента v меньше, чем медиана, и последние два больше, чем медиана.

Обратите внимание, что меня интересует, скажем, децилирование, а не только медиана!

IOW, @PaulPanzer ударил гвоздь:

np.digitize(v,np.percentile(v,quantiles)) 

спасибо!

Answer 1

Использование np.digitize:

perc = np.percentile(data, q) 
indices = np.digitize(data, perc) 

Пример q = [25,50,75], data = np.arange(8):

indices 
# array([0, 0, 1, 1, 2, 2, 3, 3]) 

Answer 2

(v > np.percentile(v, 50)).astype(int) 
Out[93]: 
array([0, 0, 1, 1])