Matlab - 获取通过一定总和限制的向量元素的索引

时间:2012-10-02 15:13:06

标签: matlab

我有一个规范化的向量(即其元素之和为1),它们按降序排序(即[0.5 0.4 0.09 0.01])。

有没有办法获得第一个元素的指数,它们的总和只是某个阈值?

例如,在我的情况下,对于0.6的阈值,我将获得10.5的索引(w.r.t。到Matlab索引向量元素的方式)。对于0.91的阈值,我将获得[1 2][0.5 0.4]的索引,依此类推。

我知道我可以通过矢量循环来做到这一点但是考虑到我使用Matlab的事实,我想问我是否可以只使用一个命令行或最大2来执行此操作,从而提高计算速度我的代码。

1 个答案:

答案 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向量为空,以防第一个元素不符合条件。