[問題] numpy broadcasting 矩陣特定位置運算

看板Python作者 (NMOSFET)時間6年前 (2018/08/25 22:34), 編輯推噓1(105)
留言6則, 3人參與, 6年前最新討論串1/1
各位大大好,小的最近又遇到一個問題, 假設有個 numpy 二維矩陣 A, 我想要讓 A 矩陣裡的值都減掉在它們位置的左上角的值, 然後放到 B 矩陣裡(左,上邊緣不計算), 例如: B[i][j] = A[i][j]-A[i-1][j-1] 這個用 for 迴圈很容易實作但速度很慢, 請問各位大大有沒有能夠把 numpy broadcasting 套用在這個問題上 ? 懇請大大能夠幫小弟解惑, 感恩!! -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 163.18.104.158 ※ 文章網址: https://www.ptt.cc/bbs/Python/M.1535207666.A.2A5.html

08/25 22:42, 6年前 , 1F

08/25 23:51, 6年前 , 2F

08/25 23:52, 6年前 , 3F
我有做一個簡單的速度比較, 你可以參考看看
08/25 23:52, 3F

08/26 00:27, 6年前 , 4F
真的很感謝樓上兩位大大的幫忙!!!!
08/26 00:27, 4F

08/26 00:56, 6年前 , 5F
還可以直接試試看迴圈用numba.jit... 快很多XD
08/26 00:56, 5F

08/26 12:49, 6年前 , 6F
謝謝C大,我研究一下!
08/26 12:49, 6F
文章代碼(AID): #1RWMZoAb (Python)
文章代碼(AID): #1RWMZoAb (Python)