兩個numpy數組的逐元素減法

Question

我有兩個numpy數組，分別表示尺寸為（327600,5）和（3,5）的'a'和'b'。 我想對兩個數組進行逐個元素減法。 我將如何進行無循環操作？

當我執行a [0]-b時，我得到了（3,5）數組。 我想得到一個（327600,3,5）數組作為減法后的結果。

Answer 1

a[:, np.newaxis] - b

您需要查看a形狀a帶有額外長度為1的軸的軸，以便使這些軸正確對齊以進行廣播。