相关疑难解决方法(0)

重新启动cumsum,如果cumsum大于值,则获取索引

说我有一段距离x=[1,2,1,3,3,2,1,5,1,1]

我想从x到达总和达到10的索引,在这种情况下,idx = [4,9]。

因此,满足条件后,cumsum重新启动。

我可以使用循环来完成此操作,但是对于大型阵列而言,循环速度很慢,我想知道是否可以以某种vectorized方式进行。

python numpy pandas

5
推荐指数
3
解决办法
172
查看次数

标签 统计

numpy ×1

pandas ×1

python ×1