我目前有一個尺寸為 40 x 3 x 3 x 5 x 1000 的 5D numpy 數(shù)組，其中尺寸分別由 axbxcxdxe 標記。我有另一個 3 x 1000 尺寸的 2D numpy 數(shù)組，其中尺寸分別由 bxe 標記。我希望從 2D 數(shù)組中減去 5D 數(shù)組。我想到的一種方法是將 2D 擴展為 5D 數(shù)組（因為 2D 數(shù)組不會因其他 3 維的所有組合而改變）。我不確定我可以使用什么數(shù)組方法/numpy 函數(shù)來執(zhí)行此操作。我傾向于開始迷失 nD 數(shù)組操作。謝謝你的幫助。
2 回答
繁花如伊
TA貢獻2012條經(jīng)驗 獲得超12個贊
In [217]: a,b,c,d,e = 2,3,4,5,6
In [218]: A = np.ones((a,b,c,d,e),int); B = np.ones((b,e),int)
In [219]: A.shape
Out[219]: (2, 3, 4, 5, 6)
In [220]: B.shape
Out[220]: (3, 6)
In [221]: B[None,:,None,None,:].shape # could also use reshape()
Out[221]: (1, 3, 1, 1, 6)
In [222]: C = B[None,:,None,None,:]-A
In [223]: C.shape
Out[223]: (2, 3, 4, 5, 6)
第一個None不是必需的；numpy 將根據(jù)需要添加它，但作為人類，它可能有助于查看它。
當年話下
TA貢獻1890條經(jīng)驗 獲得超9個贊
IIUC，假設(shè)您的數(shù)組是 a 和 b：
np.swapaxes(np.swapaxes(a, 1, 3) - b, 1, 3)
添加回答
舉報
0/150
提交
取消