大家好
我想问的问题如下：
今天有两个一维矩阵A跟B 其中A单纯有很多的值, B里面则是不同的index, 后面会用到
我们想要在介于两个index之间做cumsum, 也就是说, 我们想要从B[i]累加到B[i+1]
如果这个cumsum在任何时候小于0 那就回传1
如果如果都没有的话 那就回传0
所以最后希望是回传一个len(B) - 1 的矩阵
不过目前一直没办法向量化上面这个过程
如果可以的话 也可以直接回传总共有几个1就好
目前有看到的是
reduceat 可是只能作用在ufunc上
其他太复杂 而且目前没办法达到我想要的效果
目前是先用numba先写一个大循环
不过还是想问问看有没有办法向量化
谢谢大家