使用numpy计算加权百分位数

Question

使用numpy计算加权百分位数

pythonnumpyweightedpercentile

53

有没有一种方法可以使用numpy.percentile函数计算加权百分位数？或者是否有其他的Python函数可用于计算加权百分位数？

谢谢！

- user308827

在我看来，Sam A 在下面的解决方案（https://dev59.com/72Eh5IYBdhLWcg3w9XWn#63440143）似乎是当前最佳实践的有力竞争者。 - geotheory

12个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Joan Smith · Answer 1

不幸的是，numpy并没有内置所有加权函数，但你总可以自己组合一些。

def weight_array(ar, weights):
     zipped = zip(ar, weights)
     weighted = []
     for a, w in zipped:
         for j in range(w):
             weighted.append(a)
     return weighted


np.percentile(weight_array(ar, weights), 25)

- Luca Jokull · Answer 2

这是我的解决方案：

def my_weighted_perc(data,perc,weights=None):
    if weights==None:
        return nanpercentile(data,perc)
    else:
        d=data[(~np.isnan(data))&(~np.isnan(weights))]
        ix=np.argsort(d)
        d=d[ix]
        wei=weights[ix]
        wei_cum=100.*cumsum(wei*1./sum(wei))
        return interp(perc,wei_cum,d)

它简单地计算数据的加权CDF，然后用于估计加权百分位数。