说我有一段距离x=[1,2,1,3,3,2,1,5,1,1]。
x=[1,2,1,3,3,2,1,5,1,1]
我想从x到达总和达到10的索引,在这种情况下,idx = [4,9]。
因此,满足条件后,cumsum重新启动。
我可以使用循环来完成此操作,但是对于大型阵列而言,循环速度很慢,我想知道是否可以以某种vectorized方式进行。
vectorized
python numpy pandas
numpy ×1
pandas ×1
python ×1