[問題] python numpy 向量化

看板Python作者pinner (滅。驅蚊大師)時間4年前 (2020/04/27 21:08)推噓2(2推 0噓 1→)

留言3則, 2人參與討論串1/2 (看更多)

大家好我想問的問題如下：今天有兩個一維矩陣A跟B 其中A單純有很多的值, B裏面則是不同的index, 後面會用到我們想要在介於兩個index之間做cumsum, 也就是說, 我們想要從B[i]累加到B[i+1] 如果這個cumsum在任何時候小於0 那就回傳1 如果如果都沒有的話那就回傳0 所以最後希望是回傳一個len(B) - 1 的矩陣不過目前一直沒辦法向量化上面這個過程如果可以的話也可以直接回傳總共有幾個1就好目前有看到的是 reduceat 可是只能作用在ufunc上其他太複雜而且目前沒辦法達到我想要的效果目前是先用numba先寫一個大迴圈不過還是想問問看有沒有辦法向量化謝謝大家 -- 歡迎大家來我P2個版逛逛 hlzyzi 還蠻有趣的:D -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 129.104.210.0 (法國) ※ 文章網址: https://www.ptt.cc/bbs/Python/M.1587992927.A.314.html

推

robert09080

04/28 10:03, 4年前 , 1^F

04/28 10:03, 1^F

→

robert09080

04/28 10:03, 4年前 , 2^F

04/28 10:03, 2^F

純numpy 轉成pandas再轉回來會有效能上的差異嗎？我也想過要用ffill之類的把nan轉成每區第一個數字的cumsum後全部減下去

推

darama

04/30 16:43, 4年前 , 3^F