我有一個a大小為 (1000,100,2) 的 NumPy 陣列。
我只想從沿第一個迭代的最后一個維度中減去值。換句話說:
a[i 1,j,[x1,y1]]-a[i,j,[x0,y0]]
獲得一個b大小為 (999,100,2) 的陣列,其中最后一個維度包含:
[x1-x0, y1-y0] , etc...
我知道如何使用for回圈來做到這一點,我正在尋找更有效的“numpyzed”版本
uj5u.com熱心網友回復:
看起來你想要np.diff
a = np.random.rand(1000, 100, 2)
b = np.diff(a, axis = 0)
b.shape
Out[]: (999, 100, 2)
轉載請註明出處,本文鏈接:https://www.uj5u.com/qiye/473659.html
