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

Question

我有一个归一化的向量(即其元素的总和为 1)并且它们按降序排列(即 [0.5 0.4 0.09 0.01])。

有没有办法获取总和刚好低于某个阈值的第一个元素的索引？

例如，在我的例子中，对于 0.6 的阈值，我将获得 1， 的索引(w.r.t. 到 Matlab 索引向量元素的方式) 0.5。对于 0.91 的阈值，我将获得 [1 2]、[0.5 0.4] 的索引，等等。

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

Answer 1

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 向量为空，以防第一个元素不合格。