我有一个规范化的向量(即其元素之和为1
),它们按降序排序(即[0.5 0.4 0.09 0.01]
)。
有没有办法获得第一个元素的指数,它们的总和只是某个阈值?
例如,在我的情况下,对于0.6
的阈值,我将获得1
,0.5
的索引(w.r.t。到Matlab索引向量元素的方式)。对于0.91
的阈值,我将获得[1 2]
,[0.5 0.4]
的索引,依此类推。
我知道我可以通过矢量循环来做到这一点但是考虑到我使用Matlab的事实,我想问我是否可以只使用一个命令行或最大2来执行此操作,从而提高计算速度我的代码。
答案 0 :(得分:4)
cumsum
函数计算向量元素的累积和。这样做,然后找到阈值小于累积总和的索引将给你答案:
V = [0.5 0.4 0.09 0.01]
t = 0.6;
index = find(t < cumsum(V), 1)-1
请注意,如果index = 0
,即使第一个元素的条件也不成立。
如果您坚持要获得符合条件的元素向量,您只需使用以下内容定义范围:
indices = 1 : (find(t<cumsum(V),1)-1)
在这种情况下,indices
向量为空,以防第一个元素不符合条件。