變量的累計和直到給定的百分位數

Question

我想總結一個數組中的所有值直到給定的百分位數。 例如

import numpy as np 
a = [15, 40, 124, 282, 914, 308] 
print np.percentile(a,90) 

第90百分位〜611和累積總和直到此時才461

是否有Python中的任何功能，能做到嗎？

Answer 1

A=np.array(a) 
A[:(A<np.percentile(a, 90)).argmin()].sum() #461 

@ JoshAdel的

%%timeit 
    ...: b = np.cumsum(a) 
    ...: p90 = np.percentile(a, 90) 
    ...: b[b < p90][-1] 
    ...: 
1000 loops, best of 3: 217 µs per loop 

此：

%timeit A[:(A<np.percentile(a, 90)).argmin()].sum() 
10000 loops, best of 3: 191 µs per loop 

Answer 2

沒有，我知道的，但你可以做到這一點

import numpy as np 
from itertools import takewhile 

a = [15, 40, 124, 282, 914, 308] 
p90 = np.percentile(a,90) 
print sum(takewhile(lambda x : x < p90, a)) 

日期：

461 

Answer 3

import numpy as np 
a = np.array([15, 40, 124, 282, 914, 308]) 
b = np.cumsum(a) 
p90 = np.percentile(a, 90) 
print b[b < p90][-1] #461