[英]Replacing entries in a numpy array with their quantile index with python
我有一个带有数字的一维numpy数组,我希望每个数字都替换为它所属的分位数的索引。
这是我的五分类指数代码:
import numpy as np
def get_quintile_indices( a ):
result = np.ones( a.shape[ 0 ] ) * 4
quintiles = [
np.percentile( a, 20 ),
np.percentile( a, 40 ),
np.percentile( a, 60 ),
np.percentile( a, 80 )
]
for q in quintiles:
result -= np.less_equal( a, q ) * 1
return result
a = np.array( [ 58, 54, 98, 76, 35, 13, 62, 18, 62, 97, 44, 43 ] )
print get_quintile_indices( a )
输出:
[ 2. 2. 4. 4. 0. 0. 3. 0. 3. 4. 1. 1.]
您会看到我从一个初始化为最高可能索引的数组开始,并且每个条目的每个五分位数cutract 1都小于或等于五分位数。 有一个更好的方法吗? 内置函数,可用于将数字映射到切割点列表?
首先,我们可以一次性生成这些quintiles
-
quintiles = np.percentile( a, [20,40,60,80] )
对于获得偏移的最后一步,我们可以简单地使用np.searchsorted
,这可能是你正在寻找的内置,就像这样 -
out = np.searchsorted(quintiles, a)
或者,将循环代码直接转换为矢量化版本将使用broadcasting
,如此 -
# Use broadcasting to perform those comparisons in one go.
# Then, simply sum along the first axis and subtract from 4.
out = 4 - (quintiles[:,None] >= a).sum(0)
如果quintiles
是一个列表,我们需要将其指定为数组,然后使用broadcasting
,如下所示 -
out = 4 - (np.asarray(quintiles)[:,None] >= a).sum(0)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.