Karte Zahlen auf ihre Perzentile

Question

Ich mag würde gelten das Ergebnis von numpy.percentile zu seinem Argumente, das heißt, Karte jede Zahl in dem Eingangsvektor zu seiner Quantil.

ZB wenn v=np.array([1,2,3,4]), und ich möchte nur zwei Quantile (größer und kleiner als der Median), würde ich np.array([0,0,1,1]) mir zu sagen, dass die ersten beiden Elemente v kleiner sind als der Median und die letzten beiden sind größer als die Median.

Bitte beachte, dass ich interessiert bin, sagen wir, Dezilen, nicht nur der Median!

IOW, schlug @PaulPanzer den Nagel:

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

Dank!

Answer 1

Verwendung np.digitize:

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

Beispiel 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])