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

作者: NMOSFET (NMOSFET)   2018-08-25 22:34:23
各位大大好,小的最近又遇到一個問題,
假設有個 numpy 二維矩陣 A,
我想要讓 A 矩陣裡的值都減掉在它們位置的左上角的值,
然後放到 B 矩陣裡(左,上邊緣不計算),
例如: B[i][j] = A[i][j]-A[i-1][j-1]
這個用 for 迴圈很容易實作但速度很慢,
請問各位大大有沒有能夠把 numpy broadcasting 套用在這個問題上 ?
懇請大大能夠幫小弟解惑, 感恩!!
作者: celestialgod (天)   2018-08-25 22:42:00
作者: gmccntzx1 (o.O)   2018-08-25 23:51:00
https://repl.it/@naleraphael/subtractluarray我有做一個簡單的速度比較, 你可以參考看看
作者: NMOSFET (NMOSFET)   2018-08-26 00:27:00
真的很感謝樓上兩位大大的幫忙!!!!
作者: celestialgod (天)   2018-08-26 00:56:00
還可以直接試試看迴圈用numba.jit... 快很多XD
作者: NMOSFET (NMOSFET)   2018-08-26 12:49:00
謝謝C大,我研究一下!

Links booklink

Contact Us: admin [ a t ] ucptt.com