擅长:python、mysql、java
<p>假设<code>a</code>已排序,只需计算<strong>未超过阈值的条目数,然后添加最终的条目</p>
<pre><code>a = np.arange(6)
num = (a.cumsum() < 6).sum() + 1 # 4
</code></pre>
<p>作为一项功能:</p>
<pre><code>def cum_thresh(a, thresh):
"""
the sequence a has to be sorted
"""
return (np.cumsum(a) < thresh).sum() + 1
</code></pre>