作者:
pinner (蓮踢踢)
2020-04-28 05:08:43大家好
我想問的問題如下：
今天有兩個一維矩陣A跟B 其中A單純有很多的值, B裏面則是不同的index, 後面會用到
我們想要在介於兩個index之間做cumsum, 也就是說, 我們想要從B[i]累加到B[i+1]
如果這個cumsum在任何時候小於0 那就回傳1
如果如果都沒有的話 那就回傳0
所以最後希望是回傳一個len(B) - 1 的矩陣
不過目前一直沒辦法向量化上面這個過程
如果可以的話 也可以直接回傳總共有幾個1就好
目前有看到的是
reduceat 可是只能作用在ufunc上
其他太複雜 而且目前沒辦法達到我想要的效果
目前是先用numba先寫一個大迴圈
不過還是想問問看有沒有辦法向量化
謝謝大家