[问题] 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