作者NMOSFET (NMOSFET)
看板Python
標題[問題] numpy broadcasting 矩陣特定位置運算
時間Sat Aug 25 22:34:23 2018
各位大大好,小的最近又遇到一個問題,
假設有個 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
→ gmccntzx1: 我有做一個簡單的速度比較, 你可以參考看看 08/25 23:52
→ NMOSFET: 真的很感謝樓上兩位大大的幫忙!!!! 08/26 00:27
→ celestialgod: 還可以直接試試看迴圈用numba.jit... 快很多XD 08/26 00:56
→ NMOSFET: 謝謝C大,我研究一下! 08/26 12:49