作者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://webptt.com/cn.aspx?n=bbs/Python/M.1535207666.A.2A5.html
3F:→ gmccntzx1: 我有做一个简单的速度比较, 你可以参考看看 08/25 23:52
4F:→ NMOSFET: 真的很感谢楼上两位大大的帮忙!!!! 08/26 00:27
5F:→ celestialgod: 还可以直接试试看回圈用numba.jit... 快很多XD 08/26 00:56
6F:→ NMOSFET: 谢谢C大,我研究一下! 08/26 12:49